Spaces:

kadirnar
/

torchyolo

Sleeping

kadirnar commited on Jul 8, 2024

Commit

8320782

verified ·

1 Parent(s): b75369b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,12 +41,12 @@ def object_tracker(
 inputs = [
     gr.Image(),
-    gr.inputs.Dropdown(
         label="Model Type",
         choices=["yolov5", "yolov6", "yolov8"],
         default="yolov5",
     ),
-    gr.inputs.Dropdown(
         label="Model Path",
         choices=[
             "kadirnar/yolov5s6-v6.0",
@@ -55,12 +55,12 @@ inputs = [
         ],
         default="kadirnar/yolov5s6-v6.",
     ),
-    gr.inputs.Dropdown(
         label="Tracker Type",
         choices=["NORFAIR", "STRONGSORT", "OCSORT", "BYTETRACK", "SORT"],
         default="NORFAIR",
     ),
-    gr.inputs.Dropdown(
         label="Tracker Config Path",
         choices=[
             "tracker/norfair_track.yaml",
@@ -71,7 +71,7 @@ inputs = [
         ],
         default="tracker/norfair_track.yaml",
     ),
-    gr.inputs.Dropdown(
         label="Tracker Weight Path",
         choices=[
             "kadirnar/osnet_x0_5_imagenet",

 inputs = [
     gr.Image(),
+    gr.Dropdown(
         label="Model Type",
         choices=["yolov5", "yolov6", "yolov8"],
         default="yolov5",
     ),
+    gr.Dropdown(
         label="Model Path",
         choices=[
             "kadirnar/yolov5s6-v6.0",
         ],
         default="kadirnar/yolov5s6-v6.",
     ),
+    gr.Dropdown(
         label="Tracker Type",
         choices=["NORFAIR", "STRONGSORT", "OCSORT", "BYTETRACK", "SORT"],
         default="NORFAIR",
     ),
+    gr.Dropdown(
         label="Tracker Config Path",
         choices=[
             "tracker/norfair_track.yaml",
         ],
         default="tracker/norfair_track.yaml",
     ),
+    gr.Dropdown(
         label="Tracker Weight Path",
         choices=[
             "kadirnar/osnet_x0_5_imagenet",