Update app.py

app.py

@@ -30,8 +30,12 @@ def apply_nms(predictions, iou_threshold=0.5):
     # Perform NMS using OpenCV
     indices = cv2.dnn.NMSBoxes(boxes.tolist(), scores.tolist(), score_threshold=0.25, nms_threshold=iou_threshold)
 
-    # Convert tuple of indices to a flat NumPy array
-    indices = indices.flatten() if isinstance(indices, tuple) else indices
+    # Ensure indices are a numpy array (if returned as a tuple)
+    if isinstance(indices, tuple):
+        indices = indices[0]  # Extracting the first element which contains the indices
+
+    # Now indices is a numpy array, we can safely call .flatten()
+    indices = indices.flatten()
 
     nms_predictions = []