Update app.py

app.py

@@ -22,10 +22,12 @@ st.session_state["model_name"] = model_name
 if model_name == 'Helsinki-NLP':
     try:
         model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
+        tokenizer = AutoTokenizer.from_pretrained(model_name)
+        model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
     except EnvironmentError:
         model_name = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+        tokenizer = AutoTokenizer.from_pretrained(model_name)
+        model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 else:
     tokenizer = T5Tokenizer.from_pretrained(model_name)
     model = T5ForConditionalGeneration.from_pretrained(model_name)