Update tasks/image.py

tasks/image.py

@@ -95,7 +95,9 @@ async def evaluate_image(request: ImageEvaluationRequest):
     train_test = dataset["train"].train_test_split(test_size=request.test_size, seed=request.test_seed)
     test_dataset = train_test["test"]
 
-    model = YOLOv10("best.pt")
+    # model = YOLOv10("best.pt")
+    model = YOLOv10("best.onnx")
+    
     #device = "cuda" if torch.cuda.is_available() else "cpu"
     #model = model.to(device)