Update app.py

app.py

@@ -21,6 +21,7 @@ model = AutoModelForObjectDetection.from_pretrained("PekingU/rtdetr_r50vd_coco_o
 BOUNDING_BOX_ANNOTATOR = sv.BoundingBoxAnnotator()
 MASK_ANNOTATOR = sv.MaskAnnotator()
 LABEL_ANNOTATOR = sv.LabelAnnotator()
+TRACKER = sv.ByteTrack()
 
 
 def calculate_end_frame_index(source_video_path):
@@ -63,7 +64,7 @@ def process_video(
             detections = []
             
             detections = sv.Detections.from_transformers(results[0])
-            
+            detections = TRACKER.update_with_detections(detections)
             for label in results[0]["labels"]: 
               final_labels.append(model.config.id2label[label.item()])
             frame = annotate_image(