Spaces:
Running
Running
Commit
·
f456cb1
1
Parent(s):
5869c97
troubleshooting the detect language part
Browse files
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 |
-
|
49 |
-
# if
|
|
|
50 |
if(language == "Detect Language"):
|
51 |
-
|
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 |
|