Update main.py

main.py

@@ -1,23 +1,10 @@
 from ultralytics import YOLO
-model = YOLO('best.pt')  # load a custom model
-
-# Predict with the model
-results = model('test.jpg') 
-
-# Extracting predictions with class names and confidence levels
-predictions = results.pandas().xyxy[0][['name', 'confidence']]
-
-# Sorting predictions from highest to lowest confidence
-sorted_predictions = predictions.sort_values(by='confidence', ascending=False)
-
-# Display sorted predictions
-print(sorted_predictions)# predict on an image
 
 import asyncio
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 import requests
-
+model = YOLO('best.pt')  # load a custom model
 
 app = FastAPI()
 
@@ -34,4 +21,15 @@ app.add_middleware(
     
 @app.get("/predict")
 async def get_prediction(page: str,paginate: str):
-      return "done"
+    # Predict with the model
+    results = model('test.jpg') 
+    
+    # Extracting predictions with class names and confidence levels
+    predictions = results.pandas().xyxy[0][['name', 'confidence']]
+    
+    # Sorting predictions from highest to lowest confidence
+    sorted_predictions = predictions.sort_values(by='confidence', ascending=False)
+    
+    # Display sorted predictions
+    print(sorted_predictions)# predict on an image
+    return sorted_predictions