GreenRaptor commited on
Commit
2a9c85c
·
1 Parent(s): f3a3a4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,7 +4,7 @@ import whisper
4
  import gradio as gr
5
  from transformers import pipeline
6
 
7
- transcription_model = whisper.load_model("small.en")
8
 
9
  def transcribe(audio):
10
 
@@ -17,8 +17,8 @@ def transcribe(audio):
17
  mel = whisper.log_mel_spectrogram(audio).to(transcription_model.device)
18
 
19
  # detect the spoken language
20
- # _, probs = transcription_model.detect_language(mel)
21
- # print(f"Detected language: {max(probs, key=probs.get)}")
22
 
23
  # decode the audio
24
  options = whisper.DecodingOptions(fp16 = False)
 
4
  import gradio as gr
5
  from transformers import pipeline
6
 
7
+ transcription_model = whisper.load_model("small")
8
 
9
  def transcribe(audio):
10
 
 
17
  mel = whisper.log_mel_spectrogram(audio).to(transcription_model.device)
18
 
19
  # detect the spoken language
20
+ _, probs = transcription_model.detect_language(mel)
21
+ print(f"Detected language: {max(probs, key=probs.get)}")
22
 
23
  # decode the audio
24
  options = whisper.DecodingOptions(fp16 = False)