Update modules/translation/translation_base.py

modules/translation/translation_base.py

@@ -9,7 +9,7 @@ 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.inference import NLLBInference
 
 class TranslationBase(ABC):
     def __init__(self,
@@ -171,13 +171,16 @@ class TranslationBase(ABC):
         List[dict] with translation
         """
         try:
-            TranslationBase.cache_parameters(model_size=model_size,src_lang=src_lang,tgt_lang=tgt_lang,max_length=max_length,add_timestamp=add_timestamp)
-            TranslationBase.update_model(model_size=model_size,src_lang=src_lang,tgt_lang=tgt_lang,progress=progress)
+            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)
+            
             total_progress = len(input_list_dict)
             for index, dic in enumerate(input_list_dict):
                 progress(index / total_progress, desc="Translating..")
-                translated_text = TranslationBase.translate(dic["text"], max_length=max_length)
+                translated_text = self.translate(dic["text"], max_length=max_length)
                 dic["text"] = translated_text
 
             return input_list_dict
@@ -185,7 +188,7 @@ class TranslationBase(ABC):
         except Exception as e:
             print(f"Error: {str(e)}")
         finally:
-            TranslationBase.release_cuda_memory()
+            self.release_cuda_memory()
     
     @staticmethod
     def get_device():