Update app.py

app.py

@@ -46,8 +46,13 @@ def process_video(video_path, output_option):
     if not matched_mp4_files:
         raise Exception(f"No .mp4 video files found in directory {output_dir} after conversion for {video_name}.")
 
-    # Count the number of objects detected
-    object_count = len(results.pandas().xyxy[0])
+    # Initialize object count
+    object_count = 0
+
+    # Calculate object count based on results
+    for result in results:
+        if hasattr(result, 'boxes'):
+            object_count += len(result.boxes)
 
     # Determine the output based on user's choice
     if output_option == "Count":