aka7774 commited on
Commit
ecc74bd
1 Parent(s): a3daf02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,19 +4,19 @@ import whisper
4
  model_size = 'small'
5
  model = whisper.load_model(model_size)
6
 
7
- def speech_to_text(tmp_filename, _model_size):
8
  if model_size != _model_size:
9
  model_size = _model_size
10
  model = whisper.load_model(model_size)
11
 
12
- result = model.transcribe(tmp_filename)
13
 
14
  return result["text"]
15
 
16
  gr.Interface(
17
  fn=speech_to_text,
18
  inputs=[
19
- gr.Audio(source="microphone", type="filepath"),
20
  gr.Dropdown(choices=["tiny", "base", "small", "medium", "large"]),
21
  ],
22
  outputs="text").launch()
 
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)
11
 
12
+ result = model.transcribe(audio_file)
13
 
14
  return result["text"]
15
 
16
  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()