Update app.py

app.py

@@ -12,6 +12,7 @@ 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 = []
@@ -20,7 +21,14 @@ def apply_nms(predictions, iou_threshold=0.5):
 
     # Extract boxes, scores, and class info
     for prediction in predictions:
-        boxes.append(prediction['bbox'])
+        # Construct the bounding box from x, y, width, height
+        x = prediction['x']
+        y = prediction['y']
+        width = prediction['width']
+        height = prediction['height']
+        box = [x, y, width, height]
+
+        boxes.append(box)
         scores.append(prediction['confidence'])
         classes.append(prediction['class'])
 
@@ -50,7 +58,7 @@ def apply_nms(predictions, iou_threshold=0.5):
     for i in indices:
         nms_predictions.append({
             'class': classes[i],
-            'bbox': boxes[i],
+            'bbox': boxes[i],  # Now using the constructed box
             'confidence': scores[i]
         })