Update modules/translation/translation_base.py

modules/translation/translation_base.py

@@ -166,16 +166,14 @@ class TranslationBase(ABC):
         A List of
         List[dict] with translation
         """
-        print("Get translation...")
-        
         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)
+            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)
 
             total_progress = len(input_list_dict)
             for index, dic in enumerate(input_list_dict):
                 progress(index / total_progress, desc="Translating..")
-                translated_text = self.translate(dic["text"], max_length=max_length)
+                translated_text = translate(dic["text"], max_length=max_length)
                 dic["text"] = translated_text
 
             return input_list_dict
@@ -183,7 +181,7 @@ class TranslationBase(ABC):
         except Exception as e:
             print(f"Error: {str(e)}")
         finally:
-            self.release_cuda_memory()
+            release_cuda_memory()
     
     @staticmethod
     def get_device():