Update app.py

app.py

@@ -2,12 +2,16 @@ import os
 os.system("pip3 install cython_bbox gdown 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'")
 from torchyolo import YoloHub
 import gradio as gr
+from utils import attempt_download_from_hub
 
 """ 
 Paper Implementation
 #"kadirnar/OcSort"
 #"kadirnar/SORT"
 #"kadirnar/ByteTracker"
+#"kadirnar/strongsort"
+
+
 """
 
 def object_tracker(
@@ -16,6 +20,7 @@ def object_tracker(
     model_path: str,
     tracker_type: str,
     tracker_config_path: str,
+    StrongSort_OsNet_Path: str = None,
 ):
     model = YoloHub(
         config_path="default_config.yaml",
@@ -23,11 +28,16 @@ def object_tracker(
         model_path=model_path,
     )
 
+    if tracker_type == "STRONGSORT":
+        StrongSort_OsNet_Path = attempt_download_from_hub(StrongSort_OsNet_Path)
+
     model.predict(
         source=source,
         tracker_type=tracker_type,
+        tracker_weight_path=StrongSort_OsNet_Path,
         tracker_config_path=tracker_config_path,
     )
+
     return 'output.mp4'
        
 
@@ -63,6 +73,15 @@ inputs = [
         ],
         default="tracker/norfair_track.yaml",
     ),
+    gr.inputs.Dropdown(
+        label="Tracker Weight Path",
+        choices=[
+            "kadirnar/osnet_x0_5_imagenet",
+            "kadirnar/osnet_x1_0_imagenet",
+            "kadirnar/osnet_x0_25_imagenet"
+        ],
+        default="kadirnar/osnet_x0_5_imagenet",
+    ),
 ]
 examples = [
     [