Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,6 +12,7 @@ def inference(audio):
|
|
| 12 |
mel = whisper.log_mel_spectrogram(audio).to(model.device)
|
| 13 |
|
| 14 |
_, probs = model.detect_language(mel)
|
|
|
|
| 15 |
|
| 16 |
options = whisper.DecodingOptions(fp16 = False)
|
| 17 |
result = whisper.decode(model, mel, options)
|
|
@@ -157,12 +158,15 @@ with block:
|
|
| 157 |
)
|
| 158 |
|
| 159 |
btn = gr.Button("Transcribe")
|
|
|
|
|
|
|
|
|
|
| 160 |
text = gr.Textbox(show_label=False)
|
| 161 |
|
| 162 |
|
| 163 |
|
| 164 |
|
| 165 |
-
btn.click(inference, inputs=[audio], outputs=[text])
|
| 166 |
|
| 167 |
gr.HTML('''
|
| 168 |
<div class="footer">
|
|
|
|
| 12 |
mel = whisper.log_mel_spectrogram(audio).to(model.device)
|
| 13 |
|
| 14 |
_, probs = model.detect_language(mel)
|
| 15 |
+
lang = max(probs, key=probs.get)
|
| 16 |
|
| 17 |
options = whisper.DecodingOptions(fp16 = False)
|
| 18 |
result = whisper.decode(model, mel, options)
|
|
|
|
| 158 |
)
|
| 159 |
|
| 160 |
btn = gr.Button("Transcribe")
|
| 161 |
+
|
| 162 |
+
lang_str = gr.Textbox(show_label=False)
|
| 163 |
+
|
| 164 |
text = gr.Textbox(show_label=False)
|
| 165 |
|
| 166 |
|
| 167 |
|
| 168 |
|
| 169 |
+
btn.click(inference, inputs=[audio], outputs=[lang_str,text])
|
| 170 |
|
| 171 |
gr.HTML('''
|
| 172 |
<div class="footer">
|