Update tasks/image.py

tasks/image.py

@@ -6,6 +6,7 @@ from sklearn.metrics import accuracy_score, precision_score, recall_score
 import random
 import os
 from ultralytics import YOLO
+from torch.utils.data import DataLoader
 
 from .utils.evaluation import ImageEvaluationRequest
 from .utils.emissions import tracker, clean_emissions_data, get_space_info