Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,8 +21,8 @@ def generate(text: str):
|
|
| 21 |
return {"output": output[0]["generated_text"]}
|
| 22 |
|
| 23 |
@app.post("/uploadfile/")
|
| 24 |
-
async def create_upload_file(
|
| 25 |
-
pred, idx, probs = learn.predict(file)
|
| 26 |
return dict(zip(categories, map(float, probs)))
|
| 27 |
# pred, idx, probs = learn.predict(img)
|
| 28 |
# return dict(zip(categories, map(float, probs)))
|
|
|
|
| 21 |
return {"output": output[0]["generated_text"]}
|
| 22 |
|
| 23 |
@app.post("/uploadfile/")
|
| 24 |
+
async def create_upload_file(image: UploadFile):
|
| 25 |
+
pred, idx, probs = learn.predict(image.file)
|
| 26 |
return dict(zip(categories, map(float, probs)))
|
| 27 |
# pred, idx, probs = learn.predict(img)
|
| 28 |
# return dict(zip(categories, map(float, probs)))
|