Update app.py

app.py

@@ -252,11 +252,15 @@ def process_video(video_data, progress_callback=None):
                         continue
                     seen_violations.add(label)
 
+                    xywh_array = box.xywh.cpu().numpy()
+                    if xywh_array.shape[0] == 0:
+                        continue
+
                     violation = {
                         "frame": frame_count,
                         "violation": label,
                         "confidence": round(conf, 2),
-                        "bounding_box": [round(x, 2) for x in box.xywh.cpu().numpy()[0]],
+                        "bounding_box": [round(x, 2) for x in xywh_array[0]],
                         "timestamp": frame_count / fps
                     }
                     violations.append(violation)