freealise commited on
Commit
be68ce1
1 Parent(s): 2f95367

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -161,7 +161,7 @@ with gr.Blocks(css=css) as demo:
161
  input_video = gr.Video(label="Input Video")
162
  submit = gr.Button("Submit")
163
  with gr.Column():
164
- model_type = gr.Dropdown(["vits", "vitb", "vitl"], type="value", value="vits", label='Model Type')
165
  processed_video = gr.Video(label="Processed Video")
166
 
167
  def on_submit(uploaded_video,model_type):
 
161
  input_video = gr.Video(label="Input Video")
162
  submit = gr.Button("Submit")
163
  with gr.Column():
164
+ model_type = gr.Dropdown([("small", "vits"), ("base", "vitb"), ("large", "vitl")], type="value", value="vits", label='Model Type')
165
  processed_video = gr.Video(label="Processed Video")
166
 
167
  def on_submit(uploaded_video,model_type):