Bug fix

Fix 'UnboundLocalError: local variable 'translation_model' referenced before assignment' error.
abdeladim-s · Apr 23, 2023 · c68674d · c68674d
1 parent f179261
commit c68674d
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/subsai/main.py b/src/subsai/main.py
@@ -142,11 +142,9 @@ def available_translation_languages(model: Union[str, TranslationModel]) -> list
         :return: list of available languages
         """
         if type(model) == str:
-            translation_model = Tools.create_translation_model(model)
-            langs = translation_model.available_languages()
+            langs = Tools.create_translation_model(model).available_languages()
         else:
             langs = model.available_languages()
-        del translation_model
         return langs
 
     @staticmethod