Hugging Face's logo

Spaces:
kadirnar
/
Audio-WebUI
Sleeping

App Files Community
3
kadirnar commited on
Commit
aa4d6ec
•
1 Parent(s): 5111cb5

update

Browse files
Files changed (1) hide show
  1. app.py +52 -54
app.py CHANGED
@@ -120,61 +120,31 @@ def sahi_app():
120
  with gr.Row():
121
  with gr.Column():
122
  sahi_image_file = gr.Image(type="filepath").style(height=260)
123
-
124
  with gr.Row():
125
  with gr.Column():
126
  sahi_autoseg_model_type = gr.Dropdown(
127
- choices=[
128
- "vit_h",
129
- "vit_l",
130
- "vit_b",
131
- ],
132
- value="vit_l",
133
- label="Sam Model Type",
134
- )
135
- sahi_model_type = gr.Dropdown(
136
- choices=[
137
- "yolov5",
138
- "yolov8",
139
- ],
140
- value="yolov5",
141
- label="Detector Model Type",
142
- )
143
-
144
- sahi_model_path = gr.Dropdown(
145
- choices=[
146
- "yolov5m",
147
- "yolov5l",
148
- "yolov5m6",
149
- "yolov5l6",
150
- ],
151
- value="yolov5m",
152
- label="Detector Model Path",
153
- )
154
- sahi_conf_th = gr.Slider(
155
- minimum=0,
156
- maximum=1,
157
- step=0.1,
158
- value=0.2,
159
- label="Confidence Threshold",
160
- )
161
-
162
  sahi_image_size = gr.Slider(
163
  minimum=0,
164
  maximum=1600,
165
  step=32,
166
  value=640,
167
  label="Image Size",
168
- )
169
-
170
- sahi_slice_height = gr.Slider(
171
  minimum=0,
172
- maximum=640,
173
- step=32,
174
- value=256,
175
- label="Slice Height",
176
- )
177
-
178
  sahi_slice_width = gr.Slider(
179
  minimum=0,
180
  maximum=640,
@@ -190,14 +160,42 @@ def sahi_app():
190
  value=0.2,
191
  label="Overlap Height",
192
  )
193
-
194
- sahi_overlap_width = gr.Slider(
195
- minimum=0,
196
- maximum=1,
197
- step=0.1,
198
- value=0.2,
199
- label="Overlap Width",
200
- )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
201
 
202
 
203
  sahi_image_predict = gr.Button(value="Generator")
@@ -226,7 +224,7 @@ def sahi_app():
226
  def metaseg_app():
227
  app = gr.Blocks()
228
  with app:
229
- gr.Markdown("# **<h2 align='center'>Segment Anything + Video + Package</h2>**")
230
  gr.Markdown(
231
  """
232
  <h5 style='text-align: center'>
 
120
  with gr.Row():
121
  with gr.Column():
122
  sahi_image_file = gr.Image(type="filepath").style(height=260)
 
123
  with gr.Row():
124
  with gr.Column():
125
  sahi_autoseg_model_type = gr.Dropdown(
126
+ choices=[
127
+ "vit_h",
128
+ "vit_l",
129
+ "vit_b",
130
+ ],
131
+ value="vit_l",
132
+ label="Sam Model Type",
133
+ )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
134
  sahi_image_size = gr.Slider(
135
  minimum=0,
136
  maximum=1600,
137
  step=32,
138
  value=640,
139
  label="Image Size",
140
+ )
141
+ sahi_conf_th = gr.Slider(
 
142
  minimum=0,
143
+ maximum=1,
144
+ step=0.1,
145
+ value=0.2,
146
+ label="Confidence Threshold",
147
+ )
 
148
  sahi_slice_width = gr.Slider(
149
  minimum=0,
150
  maximum=640,
 
160
  value=0.2,
161
  label="Overlap Height",
162
  )
163
+ with gr.Row():
164
+ with gr.Column():
165
+ sahi_model_type = gr.Dropdown(
166
+ choices=[
167
+ "yolov5",
168
+ "yolov8",
169
+ ],
170
+ value="yolov5",
171
+ label="Detector Model Type",
172
+ )
173
+
174
+ sahi_model_path = gr.Dropdown(
175
+ choices=[
176
+ "yolov5m",
177
+ "yolov5l",
178
+ "yolov5m6",
179
+ "yolov5l6",
180
+ ],
181
+ value="yolov5m",
182
+ label="Detector Model Path",
183
+ )
184
+ sahi_slice_height = gr.Slider(
185
+ minimum=0,
186
+ maximum=640,
187
+ step=32,
188
+ value=256,
189
+ label="Slice Height",
190
+ )
191
+
192
+ sahi_overlap_width = gr.Slider(
193
+ minimum=0,
194
+ maximum=1,
195
+ step=0.1,
196
+ value=0.2,
197
+ label="Overlap Width",
198
+ )
199
 
200
 
201
  sahi_image_predict = gr.Button(value="Generator")
 
224
  def metaseg_app():
225
  app = gr.Blocks()
226
  with app:
227
+ gr.Markdown("# **<h2 align='center'>Segment Anything + Video + SAHI</h2>**")
228
  gr.Markdown(
229
  """
230
  <h5 style='text-align: center'>