Update app.py

app.py

@@ -57,11 +57,20 @@ def predict(image):
         scores = predictions[0]['scores'].cpu().numpy()
         labels = predictions[0]['labels'].cpu().numpy()
 
+        # Debugging: Print predictions
+        print("Boxes:", boxes)
+        print("Scores:", scores)
+        print("Labels:", labels)
+
         # Filter out low-confidence predictions
         threshold = 0.5
         boxes = boxes[scores > threshold]
         labels = labels[scores > threshold]
 
+        # Debugging: Print filtered predictions
+        print("Filtered Boxes:", boxes)
+        print("Filtered Labels:", labels)
+
         # Convert the input image to a NumPy array
         image_np = np.array(image)