Update tasks/image.py

tasks/image.py

@@ -191,7 +191,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     #--------------------------------------------------------------------------------------------   
     smoke_dataset = SmokeDataset(test_dataset,feature_extractor)
     # dataloader = DataLoader(smoke_dataset, batch_size=16, shuffle=False)
-    dataloader = DataLoader(dataset, batch_size=8, collate_fn=collate_fn)
+    dataloader = DataLoader(dataset["test"], batch_size=8, collate_fn=collate_fn)
 
     predictions = []
     true_labels = []