Update tasks/image.py

tasks/image.py

@@ -154,7 +154,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     # Update the code below to replace the random baseline with your model inference
     #--------------------------------------------------------------------------------------------   
     
-    dataloader = DataLoader(test_dataset, batch_size=batch_size, shuffle=False)
+    dataloader = DataLoader(test_dataset, batch_size=32, shuffle=False)
 
     predictions = []
     true_labels = []