Update tasks/image.py

tasks/image.py

@@ -107,7 +107,6 @@ async def evaluate_image(request: ImageEvaluationRequest):
     pred_boxes = []  # Flattened list of predicted boxes
     true_boxes_list = []  # Flattened list of ground truth boxes
 
-    print('Start of prediction...')
     for example in test_dataset:
         # Extract image and annotations
         image = example["image"]
@@ -139,12 +138,10 @@ async def evaluate_image(request: ImageEvaluationRequest):
             predictions.append(0)  # No smoke predicted
             pred_boxes.append([])  # Add empty list if no prediction
 
-    print('Prediction done.')
     # Filter out entries with empty boxes
     filtered_true_boxes_list = []
     filtered_pred_boxes = []
-    
-    print('Box analysis...')
+
     for true_boxes, pred_boxes_entry in zip(true_boxes_list, pred_boxes):
         if true_boxes and pred_boxes_entry:  # Keep only if neither is empty
             filtered_true_boxes_list.append(true_boxes)