Update app.py

app.py

@@ -10,6 +10,15 @@ model = T5ForConditionalGeneration.from_pretrained(model_name)
 # Create the app layout
 st.title("Text Translation App")
 input_text = st.text_input("Enter text to translate ENDE:")
+# Create a list of options for the select box
+options = ["English", "Romanian", "German", "French", "Spanish"]
+# Create the select box
+slanguage_selector = st.selectbox("Select a source language:", options)
+tlanguage_selector = st.selectbox("Select a target language:", options)
+# Display the selected language
+sselected_language = st.session_state["sselected_language"]
+tselected_language = st.session_state["tselected_language"]
+st.write("Selected language combination:", selected_language, ' - ', selected_language)
 submit_button = st.button("Translate")
 translated_text = st.text("")
 
@@ -18,7 +27,7 @@ if submit_button:
     # Encode the input text
     # encoded = tokenizer(text_input, return_tensors="pt")
     
-    input_ids = tokenizer.encode(f'Translate English to German: {input_text}', return_tensors='pt')
+    input_ids = tokenizer.encode(f'Translate {sselected_language} to {tselected_language]: {input_text}', return_tensors='pt')
 
     # Perform translation
     output_ids = model.generate(input_ids)