Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -65,8 +65,8 @@ def process_video(
|
|
65 |
|
66 |
detections = sv.Detections.from_transformers(results[0])
|
67 |
detections = TRACKER.update_with_detections(detections)
|
68 |
-
for label in
|
69 |
-
|
70 |
frame = annotate_image(
|
71 |
input_image=frame,
|
72 |
detections=detections,
|
|
|
65 |
|
66 |
detections = sv.Detections.from_transformers(results[0])
|
67 |
detections = TRACKER.update_with_detections(detections)
|
68 |
+
for label in detections.class_id.tolist():
|
69 |
+
final_labels.append(model.config.id2label[label])
|
70 |
frame = annotate_image(
|
71 |
input_image=frame,
|
72 |
detections=detections,
|