aka7774 commited on
Commit
3786422
·
1 Parent(s): ecc74bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,10 +1,11 @@
1
  import gradio as gr
2
  import whisper
3
 
4
- model_size = 'small'
5
  model = whisper.load_model(model_size)
6
 
7
  def speech_to_text(audio_file, _model_size):
 
8
  if model_size != _model_size:
9
  model_size = _model_size
10
  model = whisper.load_model(model_size)
@@ -17,6 +18,6 @@ gr.Interface(
17
  fn=speech_to_text,
18
  inputs=[
19
  gr.Audio(source="upload", type="filepath"),
20
- gr.Dropdown(choices=["tiny", "base", "small", "medium", "large"]),
21
  ],
22
  outputs="text").launch()
 
1
  import gradio as gr
2
  import whisper
3
 
4
+ model_size = 'large'
5
  model = whisper.load_model(model_size)
6
 
7
  def speech_to_text(audio_file, _model_size):
8
+ global model_size, model
9
  if model_size != _model_size:
10
  model_size = _model_size
11
  model = whisper.load_model(model_size)
 
18
  fn=speech_to_text,
19
  inputs=[
20
  gr.Audio(source="upload", type="filepath"),
21
+ gr.Dropdown(value=model_size, choices=["tiny", "base", "small", "medium", "large"]),
22
  ],
23
  outputs="text").launch()