Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -19,7 +19,8 @@ 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.translation_base import TranslationBase
+#from modules.translation.translation_base import TranslationBase
+from modules.translation.nllb_inference import NLLBInference
 
 class WhisperBase(ABC):
     def __init__(self,
@@ -259,8 +260,8 @@ class WhisperBase(ABC):
                 #files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext)}
 
                 # Translate the trandscribed segments
-                TranslateInstance = TranslationBase()
-                test = TranslateInstance.translate_text(
+                self.nllb_inf = NLLBInference()
+                test = self.nllb_inf.translate_text(
                     input_list_dict=transcribed_segments,
                     model_size="facebook/nllb-200-distilled-600M",
                     src_lang=file_language,