Update modules/translation/translation_base.py

modules/translation/translation_base.py

@@ -6,7 +6,7 @@ import gc
 from typing import List
 from datetime import datetime
 
-from modules.translation.nllb_inference import *
+import modules.translation.nllb_inference as nllb
 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
@@ -225,8 +225,8 @@ class TranslationBase(ABC):
                          max_length: int,
                          add_timestamp: bool):
         def validate_lang(lang: str):
-            if lang in list(self.NLLB_AVAILABLE_LANGS.values()):
-                flipped = {value: key for key, value in self.NLLB_AVAILABLE_LANGS.items()}
+            if lang in list(nllb.NLLB_AVAILABLE_LANGS.values()):
+                flipped = {value: key for key, value in nllb.NLLB_AVAILABLE_LANGS.items()}
                 return flipped[lang]
             return lang