Update tasks/image.py

tasks/image.py

@@ -102,7 +102,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     
     model_path = Path("tasks", "models")
     model_name = "best.pt"
-    model = YOLO(Path(model_path, model_name), task="detect")
+    model = YOLO(Path(model_path, model_name), task="detect",weights_only = True)
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model = model.to(device)    
     predictions = []