Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -19,7 +19,7 @@ from modules.utils.files_manager import get_media_files, format_gradio_files, lo
 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 translate
+from modules.translation.nllb_inference import NLLBInference
 
 
 
@@ -267,7 +267,7 @@ class WhisperBase(ABC):
                     translate_input += item['text'] + "\n"	
 
                 translate_input = translate_input.rstrip()
-                translated_text = translate(translate_input,200)
+                translated_text = self.nllb_inf.translate(translate_input,200)
                 print(translated_text)
                 
                 ## Get preview as txt