Update tasks/image.py

tasks/image.py

@@ -126,8 +126,9 @@ async def evaluate_image(request: ImageEvaluationRequest):
         else:
             true_boxes_list.append([])
     
-        results = yolo_model .predict(image, verbose=False)    # INFERENCE - prediction
-        
+        # results = yolo_model .predict(image, verbose=False)    # INFERENCE - prediction
+        results = yolo_model.predict(image)    # INFERENCE - prediction
+
         if len(results[0].boxes):
             pred_box = results[0].boxes.xywhn[0].cpu().numpy().tolist()
             predictions.append(1)