Spaces:

hb-setosys
/

setosys_people_in_video_model

Sleeping

hb-setosys commited on Jan 26

Commit

875bfcc

verified ·

1 Parent(s): 97bc949

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,7 @@ import gradio as gr
 from ultralytics import YOLO
 import cv2
 from deep_sort_realtime.deepsort_tracker import DeepSort
 def initialize_tracker(max_age=30, n_init=3, nn_budget=100):
     return DeepSort(max_age=max_age, n_init=n_init, nn_budget=nn_budget)

 from ultralytics import YOLO
 import cv2
 from deep_sort_realtime.deepsort_tracker import DeepSort
+import logging
 def initialize_tracker(max_age=30, n_init=3, nn_budget=100):
     return DeepSort(max_age=max_age, n_init=n_init, nn_budget=nn_budget)