chinmaydan commited on
Commit
f456cb1
·
1 Parent(s): 5869c97

troubleshooting the detect language part

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -45,12 +45,11 @@ def predict(audio, language, mic_audio=None):
45
 
46
  # Calculates the mel frequency spectogram
47
  mel = whisper.log_mel_spectrogram(audio).to(model.device)
48
- outLanguage = 'ma'
49
- # if
 
50
  if(language == "Detect Language"):
51
- lang_token, lang_probs = model.detect_language(mel)
52
- languages = [max(probs, key=probs.get) for probs in lang_probs]
53
- print("Detected language is: " + languages)
54
  else:
55
  outLanguage = language_id_lookup[language.split()[0]]
56
 
 
45
 
46
  # Calculates the mel frequency spectogram
47
  mel = whisper.log_mel_spectrogram(audio).to(model.device)
48
+
49
+ # if model is supposed to detect language, set outLanguage to None
50
+ # otherwise set to specified language
51
  if(language == "Detect Language"):
52
+ outLanguage = None
 
 
53
  else:
54
  outLanguage = language_id_lookup[language.split()[0]]
55