Update modules/translation/translation_base.py

modules/translation/translation_base.py

@@ -136,7 +136,8 @@ class TranslationBase(ABC):
         finally:
             self.release_cuda_memory()
 
-    def translate_text(input_list_dict: list,
+    def translate_text(self,
+                       input_list_dict: list,
                        model_size: str,
                        src_lang: str,
                        tgt_lang: str,
@@ -170,13 +171,13 @@ class TranslationBase(ABC):
         List[dict] with translation
         """
         try:
-            cache_parameters(model_size=model_size,src_lang=src_lang,tgt_lang=tgt_lang,max_length=max_length,add_timestamp=add_timestamp)
-            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)
 
             total_progress = len(input_list_dict)
             for index, dic in enumerate(input_list_dict):
                 progress(index / total_progress, desc="Translating..")
-                translated_text = 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
@@ -184,7 +185,7 @@ class TranslationBase(ABC):
         except Exception as e:
             print(f"Error: {str(e)}")
         finally:
-            release_cuda_memory()
+            self.release_cuda_memory()
     
     @staticmethod
     def get_device():