Update app.py

app.py

@@ -1,8 +1,6 @@
 import streamlit as st
 from transformers import T5Tokenizer, T5ForConditionalGeneration
-model_name = 't5-base'
-tokenizer = T5Tokenizer.from_pretrained(model_name)
-model = T5ForConditionalGeneration.from_pretrained(model_name)
+
 
 # model = T5ForTranslation.from_pretrained("Helsinki-NLP/opus-mt-en-ro")
 # tokenizer = T5Tokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ro")
@@ -20,6 +18,9 @@ model_selector = st.selectbox("Select a model:", models)
 # Display the selected language
 sselected_language = st.session_state["sselected_language"]
 tselected_language = st.session_state["tselected_language"]
+model_name = st.session_state["model_selector"]
+tokenizer = T5Tokenizer.from_pretrained(model_name)
+model = T5ForConditionalGeneration.from_pretrained(model_name)
 st.write("Selected language combination:", selected_language, ' - ', selected_language)
 submit_button = st.button("Translate")
 translated_text = st.text("")