jiedong-yang commited on
Commit
d743dcc
β€’
1 Parent(s): 3bea6b1
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -169,8 +169,8 @@ with demo:
169
  text = gr.Textbox(label="Transcription", placeholder="transcription")
170
 
171
  with gr.Row():
172
- default_values = dict(model='base', bs=5, bo=5) if torch.cuda.is_available() \
173
- else dict(model='tiny', bs=1, bo=1)
174
  model_options = gr.Dropdown(['Tiny', 'Base'], value=default_values['model'], label="models")
175
  model_options.change(load_model, inputs=model_options, outputs=model_options)
176
 
 
169
  text = gr.Textbox(label="Transcription", placeholder="transcription")
170
 
171
  with gr.Row():
172
+ default_values = dict(model='Base', bs=5, bo=5) if torch.cuda.is_available() \
173
+ else dict(model='Tiny', bs=1, bo=1)
174
  model_options = gr.Dropdown(['Tiny', 'Base'], value=default_values['model'], label="models")
175
  model_options.change(load_model, inputs=model_options, outputs=model_options)
176