Spaces:
Running
Running
jhj0517
commited on
Commit
·
d41e01c
1
Parent(s):
d5e37cc
Fix language key bug
Browse files
modules/whisper/base_transcription_pipeline.py
CHANGED
@@ -528,6 +528,9 @@ class BaseTranscriptionPipeline(ABC):
|
|
528 |
|
529 |
if cached_yaml["whisper"].get("lang", None) is None:
|
530 |
cached_yaml["whisper"]["lang"] = AUTOMATIC_DETECTION.unwrap()
|
|
|
|
|
|
|
531 |
|
532 |
if cached_yaml["vad"].get("max_speech_duration_s", float('inf')) == float('inf'):
|
533 |
cached_yaml["vad"]["max_speech_duration_s"] = GRADIO_NONE_NUMBER_MAX
|
|
|
528 |
|
529 |
if cached_yaml["whisper"].get("lang", None) is None:
|
530 |
cached_yaml["whisper"]["lang"] = AUTOMATIC_DETECTION.unwrap()
|
531 |
+
else:
|
532 |
+
language_dict = whisper.tokenizer.LANGUAGES
|
533 |
+
cached_yaml["whisper"]["lang"] = language_dict[cached_yaml["whisper"]["lang"]]
|
534 |
|
535 |
if cached_yaml["vad"].get("max_speech_duration_s", float('inf')) == float('inf'):
|
536 |
cached_yaml["vad"]["max_speech_duration_s"] = GRADIO_NONE_NUMBER_MAX
|