Update modules/translation/translation_base.py

modules/translation/translation_base.py

@@ -171,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)
+            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)
 
             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 = TranslationBase.translate(dic["text"], max_length=max_length)
                 dic["text"] = translated_text
 
             return input_list_dict
@@ -185,7 +185,7 @@ class TranslationBase(ABC):
         except Exception as e:
             print(f"Error: {str(e)}")
         finally:
-            release_cuda_memory()
+            TranslationBase.release_cuda_memory()
     
     @staticmethod
     def get_device():