Update app.py

app.py

@@ -12,19 +12,6 @@ LANG_CODES = {
     "Romanian":"ro"
 }
 
-if tgt == tl and src == en:
-    model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
-elif tgt == en and src == tl:
-    model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
-elif tgt == en and src == en:
-    model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
-elif tgt == tl and src == tl:
-    model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
-elif tgt == en and src == ro:
-    model = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M").to(device)
-elif tgt == ro and src == en:
-    model = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M").to(device)
-
 def translate(text, src_lang, tgt_lang, candidates:int):
     """
     Translate the text from source lang to target lang
@@ -33,6 +20,19 @@ def translate(text, src_lang, tgt_lang, candidates:int):
     src = LANG_CODES.get(src_lang)
     tgt = LANG_CODES.get(tgt_lang)
 
+    if tgt == tl and src == en:
+        model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
+    elif tgt == en and src == tl:
+        model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
+    elif tgt == en and src == en:
+        model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
+    elif tgt == tl and src == tl:
+        model = AutoModelForSeq2SeqLM.from_pretrained("Jayyydyyy/m2m100_418m_tokipona").to(device)
+    elif tgt == en and src == ro:
+        model = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M").to(device)
+    elif tgt == ro and src == en:
+        model = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M").to(device)
+
     tokenizer.src_lang = src
     tokenizer.tgt_lang = tgt