Update app.py

app.py

@@ -8,13 +8,15 @@ model = T5ForConditionalGeneration.from_pretrained(model_name)
 # tokenizer = T5Tokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ro")
 
 # Create the app layout
-st.title("Text Translation App")
-input_text = st.text_input("Enter text to translate ENDE:")
+st.title("Text Translation")
+input_text = st.text_input("Enter text to translate:")
 # Create a list of options for the select box
 options = ["English", "Romanian", "German", "French", "Spanish"]
+models = ["t5-base", "t5-small"]
 # Create the select box
 slanguage_selector = st.selectbox("Select a source language:", options)
 tlanguage_selector = st.selectbox("Select a target language:", options)
+model_selector = st.selectbox("Select a source language:", options)
 # Display the selected language
 sselected_language = st.session_state["sselected_language"]
 tselected_language = st.session_state["tselected_language"]
@@ -27,7 +29,7 @@ if submit_button:
     # Encode the input text
     # encoded = tokenizer(text_input, return_tensors="pt")
     
-    input_ids = tokenizer.encode(f'Translate {sselected_language} to {tselected_language]: {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)