TiberiuCristianLeon commited on
Commit
94e2261
·
verified ·
1 Parent(s): 043211f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -7,14 +7,20 @@ 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 source language:", options, index=0, placeholder="Select source language")
14
- tselected_language = st.selectbox("Select 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)
18
  st.session_state["sselected_language"] = sselected_language
19
  st.session_state["tselected_language"] = tselected_language
20
  st.session_state["model_name"] = model_name
 
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 = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large"]
11
+
12
+ # Create two columns
13
+ scol, tcol = st.columns(2)
14
+ # Place select boxes in columns
15
+ with scol:
16
+ sselected_language = st.selectbox("Source language:", options, index=0, placeholder="Select source language")
17
+ with tcol:
18
+ tselected_language = st.selectbox("Target language:", options, index=1, placeholder="Select target language")
19
+ model_name = st.selectbox("Select a model:", models, index=None, placeholder="Select language model")
20
 
 
 
 
21
  sl = langs[sselected_language]
22
  tl = langs[tselected_language]
23
+
24
  st.session_state["sselected_language"] = sselected_language
25
  st.session_state["tselected_language"] = tselected_language
26
  st.session_state["model_name"] = model_name