Update tasks/image.py

tasks/image.py

@@ -166,7 +166,6 @@ async def evaluate_image(request: ImageEvaluationRequest):
     
         # Model Inference
         # image_input = feature_extractor(images=image, return_tensors="pt").pixel_values
-        image_input = feature_extractor(images=image, return_tensors="pt", padding=True).pixel_values
         image_input = feature_extractor(images=[image], return_tensors="pt", padding=True).pixel_values
 
         with torch.no_grad():
@@ -184,7 +183,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
         pred_boxes.append(predicted_boxes)
     
         # Binary prediction for smoke detection
-        print(1 if len(predicted_boxes) > 0 else 0)
+        print(f"Prediction : {1 if len(predicted_boxes) > 0 else 0}")
         predictions.append(1 if len(predicted_boxes) > 0 else 0)