Update app.py

app.py

@@ -21,8 +21,10 @@ def generate(text: str):
     return {"output": output[0]["generated_text"]}
 
 @app.post("/uploadfile/")
-async def create_upload_file(image: UploadFile):
-    pred, idx, probs = learn.predict(image.file)
+async def create_upload_file(file: UploadFile):
+    request_object_content = await file.read()
+    img = Image.open(io.BytesIO(request_object_content))
+    pred, idx, probs = learn.predict(img)
     return dict(zip(categories, map(float, probs)))
 #    pred, idx, probs = learn.predict(img)
 #    return dict(zip(categories, map(float, probs)))