Update modules/translation/translation_base.py

modules/translation/translation_base.py

@@ -9,7 +9,6 @@ from modules.whisper.whisper_parameter import *
 from modules.utils.subtitle_manager import *
 from modules.utils.files_manager import load_yaml, save_yaml
 from modules.utils.paths import DEFAULT_PARAMETERS_CONFIG_PATH, NLLB_MODELS_DIR, TRANSLATION_OUTPUT_DIR
-from modules.translation.nllb_inference import NLLBInference
 
 class TranslationBase(ABC):
     def __init__(self,
@@ -172,10 +171,7 @@ class TranslationBase(ABC):
         """
         try:
             self.cache_parameters(model_size=model_size,src_lang=src_lang,tgt_lang=tgt_lang,max_length=max_length,add_timestamp=add_timestamp)
-            #self.update_model(model_size=model_size,src_lang=src_lang,tgt_lang=tgt_lang,progress=progress)
-
-            NLLBInferenceInstance = NLLBInference()
-            NLLBInferenceInstance.update_model(model_size=model_size,src_lang=src_lang,tgt_lang=tgt_lang,progress=progress)
+            self.update_model(model_size=model_size,src_lang=src_lang,tgt_lang=tgt_lang,progress=progress)
             
             total_progress = len(input_list_dict)
             for index, dic in enumerate(input_list_dict):