ISO lang code support for --load-only

182f6063f0

libretranslate/init.py

@@ -27,6 +27,8 @@ def check_and_install_models(force=False, load_only_lang_codes=None,update=False
         print("Found %s models" % len(available_packages))
         if load_only_lang_codes is not None:
             # load_only_lang_codes: List[str] (codes)
+            load_only_lang_codes = libretranslate.language.iso2model(load_only_lang_codes)
+
             # Ensure the user does not use any unavailable language code.
             unavailable_lang_codes = set(load_only_lang_codes)
             for pack in available_packages:
@@ -71,4 +73,4 @@ def check_and_install_models(force=False, load_only_lang_codes=None,update=False
         libretranslate.language.languages = translate.get_installed_languages()
         print(
             f"Loaded support for {len(translate.get_installed_languages())} languages ({len(available_packages)} models total)!"
-        )
+        )

libretranslate/language.py

@@ -14,6 +14,9 @@ aliases = {
 rev_aliases = {v.lower(): k for k, v in aliases.items()}
 
 def iso2model(lang):
+    if isinstance(lang, list):
+        return [iso2model(l) for l in lang]
+    
     if not isinstance(lang, str):
         return lang