Update app.py
Browse files
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 = ["
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
-
|
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
|