Update tasks/image.py

tasks/image.py

@@ -189,7 +189,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     # YOUR MODEL INFERENCE CODE HERE
     # Update the code below to replace the random baseline with your model inference
     #--------------------------------------------------------------------------------------------   
-    smoke_dataset = SmokeDataset(test_dataset)
+    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)