From c68674d79a49a20309b528af95c2e01cd29ec8ad Mon Sep 17 00:00:00 2001
From: woodong <woodongwong@gmail.com>
Date: Sun, 23 Apr 2023 16:34:07 +0800
Subject: [PATCH] Bug fix

Fix 'UnboundLocalError: local variable 'translation_model' referenced before assignment' error.
---
 src/subsai/main.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/subsai/main.py b/src/subsai/main.py
index b709601..14f8833 100644
--- 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