updated

app.py

@@ -4,7 +4,7 @@ import numpy as np
 from ultralytics import YOLO
 
 # Load the YOLO model
-model = YOLO('yolov5s.pt')  # You can use a different model if needed
+model = YOLO('yolov5s.pt')  # Use 'yolov5s.pt' or any YOLO model of your choice
 
 def count_people(video_file):
     count = 0
@@ -16,11 +16,12 @@ def count_people(video_file):
             break
 
         results = model(frame)
-        detections = results.pred[0]  # Get predictions
-
+        detections = results[0]  # Access the first result
+        
         # Count people detected (class ID for person is usually 0)
-        for det in detections:
-            if det[5] == 0:  # Check if class ID is 0 (person)
+        for det in detections.boxes.data:  # Access the boxes
+            class_id = int(det[5])  # Class ID is the 6th element
+            if class_id == 0:  # Check if class ID is 0 (person)
                 count += 1
 
     cap.release()