chinmaydan commited on
Commit
4121ad0
·
1 Parent(s): de6fe4c

prints language selected and the transcribed text in english

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -35,14 +35,17 @@ def predict(audio, language, mic_audio=None):
35
  audio = whisper.pad_or_trim(audio)
36
 
37
  mel = whisper.log_mel_spectrogram(audio).to(model.device)
38
- options = whisper.DecodingOptions(fp16 = False)
39
- result = whisper.decode(model, mel, options)
40
 
41
  if(language == "Detect Language"):
42
  outLanguage, probs = model.detect_language(mel)
43
  else:
44
  outLanguage = language_id_lookup[language.split()[0]]
45
 
 
 
 
 
 
46
 
47
 
48
  print(result.text + " " + outLanguage)
 
35
  audio = whisper.pad_or_trim(audio)
36
 
37
  mel = whisper.log_mel_spectrogram(audio).to(model.device)
 
 
38
 
39
  if(language == "Detect Language"):
40
  outLanguage, probs = model.detect_language(mel)
41
  else:
42
  outLanguage = language_id_lookup[language.split()[0]]
43
 
44
+ options = whisper.DecodingOptions(fp16 = False, language = outLanguage)
45
+ result = whisper.decode(model, mel, options)
46
+
47
+
48
+
49
 
50
 
51
  print(result.text + " " + outLanguage)