Update app.py
Browse files
app.py
CHANGED
@@ -5,13 +5,13 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer,
|
|
5 |
st.title("Text Translation")
|
6 |
input_text = st.text_input("Enter text to translate:")
|
7 |
# Create a list of options for the select box
|
8 |
-
options = ["
|
9 |
langs = {"English":"en", "Romanian":"ro", "German":"de", "French":"fr", "Spanish":"es"}
|
10 |
models = ["t5-base", "t5-small", "t5-large", "Helsinki-NLP"]
|
11 |
|
12 |
# Create the select box
|
13 |
-
sselected_language = st.selectbox("Select a source language:", options)
|
14 |
-
tselected_language = st.selectbox("Select a target language:", options)
|
15 |
sl = langs[sselected_language]
|
16 |
tl = langs[tselected_language]
|
17 |
model_name = st.selectbox("Select a model:", models)
|
@@ -46,5 +46,5 @@ if submit_button:
|
|
46 |
translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
47 |
# Display the translated text
|
48 |
print(translated_text)
|
49 |
-
st.write(f"Translated text from {sselected_language} to {tselected_language} using {model_name}
|
50 |
translated_textarea = st.text(translated_text)
|
|
|
5 |
st.title("Text Translation")
|
6 |
input_text = st.text_input("Enter text to translate:")
|
7 |
# Create a list of options for the select box
|
8 |
+
options = ["German", "Romanian", "English", "French", "Spanish"]
|
9 |
langs = {"English":"en", "Romanian":"ro", "German":"de", "French":"fr", "Spanish":"es"}
|
10 |
models = ["t5-base", "t5-small", "t5-large", "Helsinki-NLP"]
|
11 |
|
12 |
# Create the select box
|
13 |
+
sselected_language = st.selectbox("Select a source language:", options, index=0, placeholder="Select source language")
|
14 |
+
tselected_language = st.selectbox("Select a target language:", options, index=1, placeholder="Select target language)
|
15 |
sl = langs[sselected_language]
|
16 |
tl = langs[tselected_language]
|
17 |
model_name = st.selectbox("Select a model:", models)
|
|
|
46 |
translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
47 |
# Display the translated text
|
48 |
print(translated_text)
|
49 |
+
st.write(f"Translated text from {sselected_language} to {tselected_language} using {model_name}")
|
50 |
translated_textarea = st.text(translated_text)
|