hb-setosys commited on
Commit
2e964b5
·
verified ·
1 Parent(s): bf7d9d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import cv2
4
  from deep_sort_realtime.deepsort_tracker import DeepSort
5
 
6
  # Initialize YOLO model
7
- model = YOLO("yolov8l.pt") # Load YOLOv8 model
8
  tracker = DeepSort(max_age=30, n_init=3, nn_budget=100)
9
 
10
  def count_people_in_video(video_path):
 
4
  from deep_sort_realtime.deepsort_tracker import DeepSort
5
 
6
  # Initialize YOLO model
7
+ model = YOLO("setosys_ppl_in_video_v1.pt") # Load model
8
  tracker = DeepSort(max_age=30, n_init=3, nn_budget=100)
9
 
10
  def count_people_in_video(video_path):