Spaces:
Sleeping
Sleeping
Commit
·
4121ad0
1
Parent(s):
de6fe4c
prints language selected and the transcribed text in english
Browse files
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)
|