Update tasks/image.py

tasks/image.py

@@ -90,7 +90,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     dataset = load_dataset(request.dataset_name, token=os.getenv("HF_TOKEN"))
     
     # Split dataset
-    train_test = dataset["train"].train_test_split(test_size=0.02, seed=request.test_seed)
+    train_test = dataset["train"].train_test_split(test_size=request.test_size, seed=request.test_seed)
     test_dataset = train_test["test"]
     
     # Start tracking emissions