dgbkn commited on
Commit
e40bb12
·
1 Parent(s): 7e7d725
Files changed (1) hide show
  1. main.py +6 -1
main.py CHANGED
@@ -95,4 +95,9 @@ async def predict_wheat(image: UploadFile = File(...), model_id: str = "grian/1"
95
  predicted_image_str = base64.b64encode(buffer).decode('utf-8')
96
 
97
 
98
- return JSONResponse(content={"message": message_to_send, "count": predicted_count, "predicted_image": predicted_image_str})
 
 
 
 
 
 
95
  predicted_image_str = base64.b64encode(buffer).decode('utf-8')
96
 
97
 
98
+ return JSONResponse(content={"message": message_to_send, "count": predicted_count, "predicted_image": predicted_image_str})
99
+
100
+
101
+ @app.get("/")
102
+ async def home():
103
+ return HTMLResponse(content="<html><head><meta http-equiv='refresh' content='0; url=/index.html'></head></html>")