Update tasks/image.py

tasks/image.py

@@ -92,7 +92,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     
     # Split dataset
     train_test = dataset["train"].train_test_split(test_size=request.test_size, seed=request.test_seed)
-    test_dataset = dataset["test"]#train_test["test"]
+    test_dataset = dataset["val"]#train_test["test"]
     
     # Start tracking emissions
     tracker.start()