kadirnar commited on
Commit
8320782
1 Parent(s): b75369b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -41,12 +41,12 @@ def object_tracker(
41
 
42
  inputs = [
43
  gr.Image(),
44
- gr.inputs.Dropdown(
45
  label="Model Type",
46
  choices=["yolov5", "yolov6", "yolov8"],
47
  default="yolov5",
48
  ),
49
- gr.inputs.Dropdown(
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.inputs.Dropdown(
59
  label="Tracker Type",
60
  choices=["NORFAIR", "STRONGSORT", "OCSORT", "BYTETRACK", "SORT"],
61
  default="NORFAIR",
62
  ),
63
- gr.inputs.Dropdown(
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.inputs.Dropdown(
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",