Fix language key bug

jhj0517 · Nov 1, 2024 · d5e37cc · d5e37cc
1 parent 71e7964
commit d5e37cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/whisper/base_transcription_pipeline.py b/modules/whisper/base_transcription_pipeline.py
@@ -492,7 +492,7 @@ def validate_gradio_values(params: TranscriptionPipelineParams):
             params.whisper.lang = None
         else:
             language_code_dict = {value: key for key, value in whisper.tokenizer.LANGUAGES.items()}
-            params.whisper.lang = language_code_dict[params.lang]
+            params.whisper.lang = language_code_dict[params.whisper.lang]
 
         if params.whisper.initial_prompt == GRADIO_NONE_STR:
             params.whisper.initial_prompt = None