Update main.py

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi import FastAPI, File, UploadFile, HTTPException, Query
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel, Field
 from typing import List, Dict, Tuple, Optional
@@ -25,7 +25,7 @@ class ModelManager:
         self.model = None
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         
-    def load_models(self) -> bool:
+    def load_models(self):
         """Initialize all required models"""
         try:
             # Load YOLO model
@@ -163,8 +163,8 @@ async def startup_event():
 )
 async def process_image(
     image_file: UploadFile = File(...),
-    box_threshold: float = Field(0.05, ge=0, le=1),
-    iou_threshold: float = Field(0.1, ge=0, le=1)
+    box_threshold: float = Query(0.05, ge=0, le=1),
+    iou_threshold: float = Query(0.1, ge=0, le=1)
 ):
     """
     Process an uploaded image file and return the results.