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,
|