aadnk commited on
Commit
fcf2109
1 Parent(s): 479b187

Add large-v1 and large-v2 models

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -301,7 +301,7 @@ def create_ui(input_audio_max_duration, share=False, server_name: str = None, se
301
  ui_article = "Read the [documentation here](https://huggingface.co/spaces/aadnk/whisper-webui/blob/main/docs/options.md)"
302
 
303
  demo = gr.Interface(fn=ui.transcribe_webui, description=ui_description, article=ui_article, inputs=[
304
- gr.Dropdown(choices=["tiny", "base", "small", "medium", "large"], value=default_model_name, label="Model"),
305
  gr.Dropdown(choices=sorted(LANGUAGES), label="Language"),
306
  gr.Text(label="URL (YouTube, etc.)"),
307
  gr.Audio(source="upload", type="filepath", label="Upload Audio"),
 
301
  ui_article = "Read the [documentation here](https://huggingface.co/spaces/aadnk/whisper-webui/blob/main/docs/options.md)"
302
 
303
  demo = gr.Interface(fn=ui.transcribe_webui, description=ui_description, article=ui_article, inputs=[
304
+ gr.Dropdown(choices=["tiny", "base", "small", "medium", "large", "large-v1", "large-v2"], value=default_model_name, label="Model"),
305
  gr.Dropdown(choices=sorted(LANGUAGES), label="Language"),
306
  gr.Text(label="URL (YouTube, etc.)"),
307
  gr.Audio(source="upload", type="filepath", label="Upload Audio"),