Update main.py

main.py

@@ -1,6 +1,7 @@
 from ultralytics import YOLO
 
 import asyncio
+from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 import requests
@@ -20,11 +21,16 @@ app.add_middleware(
 
 names = {0: 'breaking-stage', 1: 'half-riping-stage', 2: 'overripe un-healthy', 3: 'ripe', 4: 'ripe_with_consumable_disease', 5: 'unripe'}   
 @app.get("/predict")
-async def get_prediction():
+async def get_prediction(file: UploadFile = File(...)):
+        
+    file_name = file.filename
+    print(file_name)
+    with open(file_name, "wb") as file_object:
+        file_object.write(file.file.read())    
     # Predict with the model
-    results = model('test.jpg') 
+    results = model(file_name) 
     # View results
     for r in results:
-        print(names[int(r.probs['top1'])]) 
+        print(names[int(r.probs.top1)]) 
 
     return r.probs