Update app.py

app.py

@@ -25,6 +25,7 @@ def preprocess_image(image):
     try:
         image = image.convert("RGB")  # Ensure image is in RGB mode
         image_tensor = F.to_tensor(image)
+        print(f"Preprocessed image tensor: {image_tensor.shape}")
         return image_tensor.unsqueeze(0).to(device)
     except Exception as e:
         print(f"Error in preprocessing image: {e}")
@@ -55,7 +56,12 @@ def detect_objects(image):
         return image
     try:
         outputs = model(image_tensor)[0]  # Get the first element of the output
+        print(f"Model raw outputs: {outputs}")
         outputs = non_max_suppression(outputs, conf_thres=0.25, iou_thres=0.45)[0]  # Apply NMS
+        if outputs is None or len(outputs) == 0:
+            print("No objects detected.")
+            return image
+        print(f"Filtered outputs: {outputs}")
         result_image = draw_boxes(image, outputs)
         return result_image
     except Exception as e: