Spaces:
Runtime error
Runtime error
Update detect_deepsort.py
Browse files- detect_deepsort.py +1 -1
detect_deepsort.py
CHANGED
@@ -111,7 +111,7 @@ def draw_boxes(frame, bbox_xyxy, draw_trails, identities=None, categories=None,
|
|
111 |
return frame
|
112 |
|
113 |
#@spaces.GPU(duration=120)
|
114 |
-
|
115 |
def run_deepsort(
|
116 |
weights=ROOT / 'yolo.pt', # model path or triton URL
|
117 |
source=ROOT / 'data/images', # file/dir/URL/glob/screen/0(webcam)
|
|
|
111 |
return frame
|
112 |
|
113 |
#@spaces.GPU(duration=120)
|
114 |
+
@smart_inference_mode()
|
115 |
def run_deepsort(
|
116 |
weights=ROOT / 'yolo.pt', # model path or triton URL
|
117 |
source=ROOT / 'data/images', # file/dir/URL/glob/screen/0(webcam)
|