Stefano5699 commited on
Commit
1c6f616
·
verified ·
1 Parent(s): d3c6267

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def generate(text: str):
22
 
23
  @app.post("/uploadfile/")
24
  async def create_upload_file(file: UploadFile):
25
- pred, idx, probs = learn.predict(img)
26
  return dict(zip(categories, map(float, probs)))
27
  # pred, idx, probs = learn.predict(img)
28
  # return dict(zip(categories, map(float, probs)))
 
22
 
23
  @app.post("/uploadfile/")
24
  async def create_upload_file(file: UploadFile):
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)))