Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -21,8 +21,6 @@ from modules.diarize.diarizer import Diarizer
 from modules.vad.silero_vad import SileroVAD
 from modules.translation.nllb_inference import NLLBInference
 
-
-
 class WhisperBase(ABC):
     def __init__(self,
                  model_dir: str = WHISPER_MODELS_DIR,
@@ -267,7 +265,7 @@ class WhisperBase(ABC):
                     translate_input += item['text'] + "\n"	
 
                 translate_input = translate_input.rstrip()
-                translated_text = self.nllb_inf.translate(translate_input,200)
+                translated_text = self.NLLBInference.translate(translate_input,200)
                 print(translated_text)
                 
                 ## Get preview as txt