Update main.py

main.py

@@ -25,7 +25,7 @@ class ModelManager:
         self.model = None
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         
-    def load_models(self):
+    def load_models(self) -> bool:
         """Initialize all required models"""
         try:
             # Load YOLO model
@@ -212,4 +212,4 @@ async def process_image(
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    uvicorn.run(app, host="0.0.0.0", port=8000)