Spaces:
Runtime error
Runtime error
jiedong-yang
commited on
Commit
β’
d743dcc
1
Parent(s):
3bea6b1
update
Browse files
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='
|
173 |
-
else dict(model='
|
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 |
|