Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -41,12 +41,12 @@ def object_tracker(
|
|
| 41 |
|
| 42 |
inputs = [
|
| 43 |
gr.Image(),
|
| 44 |
-
gr.
|
| 45 |
label="Model Type",
|
| 46 |
choices=["yolov5", "yolov6", "yolov8"],
|
| 47 |
default="yolov5",
|
| 48 |
),
|
| 49 |
-
gr.
|
| 50 |
label="Model Path",
|
| 51 |
choices=[
|
| 52 |
"kadirnar/yolov5s6-v6.0",
|
|
@@ -55,12 +55,12 @@ inputs = [
|
|
| 55 |
],
|
| 56 |
default="kadirnar/yolov5s6-v6.",
|
| 57 |
),
|
| 58 |
-
gr.
|
| 59 |
label="Tracker Type",
|
| 60 |
choices=["NORFAIR", "STRONGSORT", "OCSORT", "BYTETRACK", "SORT"],
|
| 61 |
default="NORFAIR",
|
| 62 |
),
|
| 63 |
-
gr.
|
| 64 |
label="Tracker Config Path",
|
| 65 |
choices=[
|
| 66 |
"tracker/norfair_track.yaml",
|
|
@@ -71,7 +71,7 @@ inputs = [
|
|
| 71 |
],
|
| 72 |
default="tracker/norfair_track.yaml",
|
| 73 |
),
|
| 74 |
-
gr.
|
| 75 |
label="Tracker Weight Path",
|
| 76 |
choices=[
|
| 77 |
"kadirnar/osnet_x0_5_imagenet",
|
|
|
|
| 41 |
|
| 42 |
inputs = [
|
| 43 |
gr.Image(),
|
| 44 |
+
gr.Dropdown(
|
| 45 |
label="Model Type",
|
| 46 |
choices=["yolov5", "yolov6", "yolov8"],
|
| 47 |
default="yolov5",
|
| 48 |
),
|
| 49 |
+
gr.Dropdown(
|
| 50 |
label="Model Path",
|
| 51 |
choices=[
|
| 52 |
"kadirnar/yolov5s6-v6.0",
|
|
|
|
| 55 |
],
|
| 56 |
default="kadirnar/yolov5s6-v6.",
|
| 57 |
),
|
| 58 |
+
gr.Dropdown(
|
| 59 |
label="Tracker Type",
|
| 60 |
choices=["NORFAIR", "STRONGSORT", "OCSORT", "BYTETRACK", "SORT"],
|
| 61 |
default="NORFAIR",
|
| 62 |
),
|
| 63 |
+
gr.Dropdown(
|
| 64 |
label="Tracker Config Path",
|
| 65 |
choices=[
|
| 66 |
"tracker/norfair_track.yaml",
|
|
|
|
| 71 |
],
|
| 72 |
default="tracker/norfair_track.yaml",
|
| 73 |
),
|
| 74 |
+
gr.Dropdown(
|
| 75 |
label="Tracker Weight Path",
|
| 76 |
choices=[
|
| 77 |
"kadirnar/osnet_x0_5_imagenet",
|