JasonFinley0821 commited on
Commit
c7d0a9b
·
1 Parent(s): c54a050

feat : add error message

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -112,4 +112,8 @@ async def predict(file: UploadFile = File(...)):
112
  return StreamingResponse(img_byte_arr, media_type="image/png")
113
 
114
  except Exception as e:
115
- return JSONResponse({"status": "error", "message": str(e)}, status_code=500)
 
 
 
 
 
112
  return StreamingResponse(img_byte_arr, media_type="image/png")
113
 
114
  except Exception as e:
115
+ import traceback
116
+ traceback.print_exc()
117
+ return JSONResponse(
118
+ {"status": "error", "message": str(e)}, status_code=500
119
+ )