Update app.py

app.py

@@ -18,12 +18,14 @@ model_name = st.selectbox("Select a model:", models)
 st.session_state["sselected_language"] = sselected_language
 st.session_state["tselected_language"] = tselected_language
 st.session_state["model_name"] = model_name
-tokenizer = T5Tokenizer.from_pretrained(model_name)
-model = T5ForConditionalGeneration.from_pretrained(model_name)
+
 if model_name == 'Helsinki-NLP':
     model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
     tokenizer = AutoTokenizer.from_pretrained(f"Helsinki-NLP/opus-mt-{sl}-{tl}")
     model = AutoModelForSeq2SeqLM.from_pretrained(f"Helsinki-NLP/opus-mt-{sl}-{tl}")
+else:
+    tokenizer = T5Tokenizer.from_pretrained(model_name)
+    model = T5ForConditionalGeneration.from_pretrained(model_name)
 st.write("Selected language combination:", sselected_language, " - ", tselected_language, "Selected model:", model_name)
 submit_button = st.button("Translate")
 translated_textarea = st.text("")