Rijgersberg commited on
Commit
7b66ddd
·
1 Parent(s): bcf2709
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,7 +11,7 @@ languages = {'auto': None} | {long_name: short_name for short_name, long_name in
11
 
12
 
13
 
14
- def inference(audio, language):
15
  audio = whisper.load_audio(audio)
16
  audio = whisper.pad_or_trim(audio)
17
 
@@ -19,7 +19,7 @@ def inference(audio, language):
19
 
20
  #_, probs = model.detect_language(mel)
21
 
22
- options = whisper.DecodingOptions(fp16 = False, language=language)
23
  result = whisper.decode(model, mel, options)
24
 
25
  print(result.text)
@@ -157,7 +157,7 @@ with block:
157
  share_button = gr.Button("Share to community", elem_id="share-btn", visible=False)
158
 
159
 
160
- btn.click(inference, inputs=[audio, languages[language_long_name.value]], outputs=[text, community_icon, loading_icon, share_button])
161
  share_button.click(None, [], [], _js=share_js)
162
 
163
  gr.HTML('''
 
11
 
12
 
13
 
14
+ def inference(audio, language_long_name):
15
  audio = whisper.load_audio(audio)
16
  audio = whisper.pad_or_trim(audio)
17
 
 
19
 
20
  #_, probs = model.detect_language(mel)
21
 
22
+ options = whisper.DecodingOptions(fp16 = False, language=languages[language_long_name])
23
  result = whisper.decode(model, mel, options)
24
 
25
  print(result.text)
 
157
  share_button = gr.Button("Share to community", elem_id="share-btn", visible=False)
158
 
159
 
160
+ btn.click(inference, inputs=[audio, language_long_name], outputs=[text, community_icon, loading_icon, share_button])
161
  share_button.click(None, [], [], _js=share_js)
162
 
163
  gr.HTML('''