Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

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