hb-setosys commited on
Commit
2ce9d59
·
verified ·
1 Parent(s): 36dd784

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("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):
 
4
  from deep_sort_realtime.deepsort_tracker import DeepSort
5
 
6
  # Initialize YOLO model
7
+ model = YOLO("setosys_ppl_in_video_small_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):