Update app.py

app.py

@@ -57,14 +57,14 @@ def detect_objects(image):
     if image_tensor is None:
         return image
     try:
-        outputs = model(image_tensor)[0]  # Get the first element of the output
+        outputs = model(image_tensor)
         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:
+        outputs = non_max_suppression(outputs, conf_thres=0.25, iou_thres=0.45)
+        if outputs is None or len(outputs[0]) == 0:
             print("No objects detected.")
             return image
-        print(f"Filtered outputs: {outputs}")
-        result_image = draw_boxes(image, outputs.cpu().numpy())
+        print(f"Filtered outputs: {outputs[0]}")
+        result_image = draw_boxes(image, outputs[0].cpu().numpy())
         return result_image
     except Exception as e:
         print(f"Error in detecting objects: {e}")