Update app.py

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, UploadFile
 from transformers import pipeline
 
 
@@ -7,7 +7,7 @@ from transformers import pipeline
 app = FastAPI(docs_url="/")
 
 pipe = pipeline("text2text-generation", model="google/flan-t5-small")
-
+categories = ('Heart', 'Oblong', 'Oval', 'Round', 'Square')
 
 @app.get("/generate")
 def generate(text: str):
@@ -17,4 +17,10 @@ def generate(text: str):
     can be found [here](https://huggingface.co/google/flan-t5-small).
     """
     output = pipe(text)
-    return {"output": output[0]["generated_text"]}
+    return {"output": output[0]["generated_text"]}
+
+@app.post("/uploadfile/")
+async def create_upload_file(file: UploadFile):
+    return {"test":"test123"}
+#    pred, idx, probs = learn.predict(img)
+#    return dict(zip(categories, map(float, probs)))