Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -271,7 +271,7 @@ class WhisperBase(ABC):
                 # Translate the transcribed segments
                 if translate_output:
                     self.nllb_inf = NLLBInference()
-                    if src_lang in self.nllb_inf.NLLB_AVAILABLE_LANGS.keys():
+                    if file_language in self.nllb_inf.NLLB_AVAILABLE_LANGS.keys():
                         transcribed_segments = self.nllb_inf.translate_text(
                             input_list_dict=transcribed_segments,
                             model_size=translate_model,
@@ -279,9 +279,9 @@ class WhisperBase(ABC):
                             tgt_lang=target_lang,
                             speaker_diarization=params.is_diarize
                         )
-                        print (src_lang + " found") 
+                        print (file_language + " found") 
                     else:
-                        print (src_lang + " not supported") 
+                        print (file_language + " not supported") 
                 
                 ## Get preview as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))