ncardian commited on
Commit
95e89a0
·
verified ·
1 Parent(s): ae8bef9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ async def predict(file: UploadFile = File(...)):
80
  prediction = prediction.tolist()
81
  prediction[0][0] = (prediction[0][0]/150)*height
82
 
83
- return JSONResponse(content={"prediction": prediction)
84
 
85
  except Exception as e:
86
  raise HTTPException(status_code=500, detail=f"Error processing image: {str(e)}")
 
80
  prediction = prediction.tolist()
81
  prediction[0][0] = (prediction[0][0]/150)*height
82
 
83
+ return JSONResponse(content={"prediction": prediction})
84
 
85
  except Exception as e:
86
  raise HTTPException(status_code=500, detail=f"Error processing image: {str(e)}")