Update app.py

app.py

@@ -12,8 +12,6 @@ rf = Roboflow(api_key="Otg64Ra6wNOgDyjuhMYU")
 project = rf.workspace("alat-pelindung-diri").project("nescafe-4base")
 model = project.version(16).model
 
-# Apply NMS (Non-Maximum Suppression)
-# Apply NMS (Non-Maximum Suppression)
 def apply_nms(predictions, iou_threshold=0.5):
     boxes = []
     scores = []
@@ -43,15 +41,12 @@ def apply_nms(predictions, iou_threshold=0.5):
     print(f"Indices after NMS: {indices}")
 
     # Check if indices is empty or invalid
-    if not indices or not isinstance(indices, tuple) or len(indices) == 0:
+    if indices is None or len(indices) == 0:
         print("No valid indices returned from NMS.")
         return []  # Return an empty list if no valid indices are found
 
-    # Extract the indices if they are valid
-    indices = indices[0]  # Extracting the first element which contains the indices
-
     # Flatten indices array (if returned as a tuple)
-    indices = indices.flatten() if isinstance(indices, np.ndarray) else np.array(indices).flatten()
+    indices = indices.flatten()
 
     nms_predictions = []