Update app.py

app.py

@@ -10,13 +10,13 @@ input_text = st.text_input("Enter text to translate:")
 options = ["English", "Romanian", "German", "French", "Spanish"]
 models = ["t5-base", "t5-small", "opus-mt-en-ro"]
 # Create the select box
-slanguage_selector = st.selectbox("Select a source language:", options)
-tlanguage_selector = st.selectbox("Select a target language:", options)
+sselected_language = st.selectbox("Select a source language:", options)
+tselected_language = st.selectbox("Select a target language:", options)
 model_selector = st.selectbox("Select a model:", models)
 # Display the selected language
-sselected_language = st.session_state["slanguage_selector"]
-tselected_language = st.session_state["tlanguage_selector"]
-model_name = st.session_state["model_selector"]
+st.session_state["slanguage_selector"] = sselected_language
+st.session_state["tlanguage_selector"] = tselected_language
+st.session_state["model_name"] = model_selector
 tokenizer = T5Tokenizer.from_pretrained(model_name)
 model = T5ForConditionalGeneration.from_pretrained(model_name)
 st.write("Selected language combination:", selected_language, ' - ', selected_language)