Update app.py

app.py

@@ -1,8 +1,6 @@
 import streamlit as st
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
-
-# model = T5ForTranslation.from_pretrained("Helsinki-NLP/opus-mt-en-ro")
 # tokenizer = T5Tokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ro")
 
 # Create the app layout
@@ -10,7 +8,7 @@ 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"]
+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)