Update tasks/image.py

tasks/image.py

@@ -107,9 +107,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     pred_boxes = []  # Flattened list of predicted boxes
     true_boxes_list = []  # Flattened list of ground truth boxes
     
-    i = 0
     for example in test_dataset:
-        i += 1
         # Extract image and annotations
         image = example["image"]
         annotation = example.get("annotations", "").strip()
@@ -139,9 +137,6 @@ async def evaluate_image(request: ImageEvaluationRequest):
         else:
             predictions.append(0)  # No smoke predicted
             pred_boxes.append([])  # Add empty list if no prediction
-    
-        if i == 10:
-            break
 
             
     # Filter out entries with empty boxes