merve HF staff commited on
Commit
80811d8
1 Parent(s): 120b633

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 results[0]["labels"]:
69
- final_labels.append(model.config.id2label[label.item()])
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,