Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -20,6 +20,7 @@ from modules.whisper.whisper_parameter import *
 from modules.diarize.diarizer import Diarizer
 from modules.vad.silero_vad import SileroVAD
 from modules.translation.nllb_inference import NLLBInference
+from modules.translation.nllb_inference import NLLB_AVAILABLE_LANGS 
 
 class WhisperBase(ABC):
     def __init__(self,
@@ -271,7 +272,7 @@ class WhisperBase(ABC):
                 # Translate the transcribed segments
                 if translate_output:
                     self.nllb_inf = NLLBInference()
-                    if file_language in self.nllb_inf.NLLB_AVAILABLE_LANGS.keys():
+                    if file_language in self.NLLB_AVAILABLE_LANGS.keys():
                         transcribed_segments = self.nllb_inf.translate_text(
                             input_list_dict=transcribed_segments,
                             model_size=translate_model,