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