TiberiuCristianLeon commited on
Commit
4bd7b04
·
verified ·
1 Parent(s): 7c8737a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 = ["English", "Romanian", "German", "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)
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)