Update tasks/image.py

tasks/image.py

@@ -182,7 +182,7 @@ async def evaluate_image(model_path: str = "models_v3/yolo11s_best.engine", requ
     dataset = load_dataset(request.dataset_name, token=os.getenv("HF_TOKEN"))
     
     # Split dataset
-    train_test = dataset["train"]
+    # train_test = dataset["train"]
     test_dataset = dataset["val"]
     
     model = YOLO(model_path, task="detect")