Update app.py

app.py

@@ -187,6 +187,7 @@ examples_images = ['data/images/1.jpg',
 examples_videos = ['data/video/1.mp4','data/video/2.mp4'] 
 
 models = ['yolov8m','yolov7','yolov7t']
+trackers = ['strongsort', 'bytetrack', 'ocsort']
 
 with gr.Blocks() as demo:
     gr.Markdown("## IDD Inference on Yolo V7 and V8 ")
@@ -208,6 +209,7 @@ with gr.Blocks() as demo:
             video_output = gr.Video(type="pil", label="Output Video",format="mp4")
         fps_video = gr.Number(0,label='FPS')
         video_drop = gr.Dropdown(choices=models,value=models[0])
+        tracking_drop = gr.Dropdown(choices=trackers,value=models[0])
         video_iou_threshold = gr.Slider(label="IOU Threshold",interactive=True, minimum=0.0, maximum=1.0, value=0.45)
         video_conf_threshold = gr.Slider(label="Confidence Threshold",interactive=True, minimum=0.0, maximum=1.0, value=0.25)
         gr.Examples(examples=examples_videos,inputs=video_input,outputs=video_output)
@@ -223,7 +225,7 @@ with gr.Blocks() as demo:
     video_button_detect.click(inference2, inputs=[video_input,video_drop,
                                            video_iou_threshold,video_conf_threshold],            
                                         outputs=[video_output,fps_video])
-    video_button_track.click(MODT,inputs=[video_input,video_drop, 'strongsort'],          
+    video_button_track.click(MODT,inputs=[video_input,video_drop, tracking_drop],          
                              outputs=[video_output, fps_video])
 
 demo.launch(debug=True,enable_queue=True)