Update app.py

app.py

@@ -7,13 +7,13 @@ input_text = st.text_input("Enter text to translate:")
 # Create a list of options for the select box
 options = ["English", "Romanian", "German", "French", "Spanish"]
 langs = {"English":"en", "Romanian":"ro", "German":"de", "French":"fr", "Spanish":"sp"}
-models = ["t5-base", "t5-small", "t5-large", "google/mt5-base", "Helsinki-NLP"]
+models = ["t5-base", "t5-small", "t5-large", "Helsinki-NLP"]
 
 # Create the select box
 sselected_language = st.selectbox("Select a source language:", options)
 tselected_language = st.selectbox("Select a target language:", options)
-sl = langs['sselected_language']  
-tl = langs['tselected_language']
+sl = langs[sselected_language]  
+tl = langs[tselected_language]
 model_name = st.selectbox("Select a model:", models)
 st.session_state["sselected_language"] = sselected_language
 st.session_state["tselected_language"] = tselected_language