Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,6 @@ models = ["t5-base", "t5-small", "opus-mt-en-ro"]
|
|
13 |
sselected_language = st.selectbox("Select a source language:", options)
|
14 |
tselected_language = st.selectbox("Select a target language:", options)
|
15 |
model_name = st.selectbox("Select a model:", models)
|
16 |
-
# Display the selected language
|
17 |
st.session_state["sselected_language"] = sselected_language
|
18 |
st.session_state["tselected_language"] = tselected_language
|
19 |
st.session_state["model_name"] = model_name
|
@@ -25,7 +24,7 @@ translated_textarea = st.text("")
|
|
25 |
|
26 |
# Handle the submit button click
|
27 |
if submit_button:
|
28 |
-
prompt = f'translate {sselected_language} to {tselected_language}: {input_text}'
|
29 |
print(prompt)
|
30 |
input_ids = tokenizer.encode(prompt, return_tensors='pt')
|
31 |
# Perform translation
|
|
|
13 |
sselected_language = st.selectbox("Select a source language:", options)
|
14 |
tselected_language = st.selectbox("Select a target language:", options)
|
15 |
model_name = st.selectbox("Select a model:", models)
|
|
|
16 |
st.session_state["sselected_language"] = sselected_language
|
17 |
st.session_state["tselected_language"] = tselected_language
|
18 |
st.session_state["model_name"] = model_name
|
|
|
24 |
|
25 |
# Handle the submit button click
|
26 |
if submit_button:
|
27 |
+
prompt = f'Prompt: translate {sselected_language} to {tselected_language}: {input_text}'
|
28 |
print(prompt)
|
29 |
input_ids = tokenizer.encode(prompt, return_tensors='pt')
|
30 |
# Perform translation
|