Update app.py

app.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI, UploadFile
 from transformers import pipeline
 from fastai.vision.all import *
+from PIL import Image
 
 # NOTE - we configure docs_url to serve the interactive Docs at the root path
 # of the app. This way, we can use the docs as a landing page for the app on Spaces.
@@ -21,10 +22,8 @@ def generate(text: str):
     return {"output": output[0]["generated_text"]}
 
 @app.post("/uploadfile/")
-async def create_upload_file(file: UploadFile):
+async def create_upload_file(file: UploadFile = File(...)):
     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)))
+    return dict(zip(categories, map(float, probs)))