Sabina Muratova
commited on
Commit
·
d9c64f9
1
Parent(s):
fd459ff
Add application file
Browse files
app.py
CHANGED
|
@@ -1,11 +1,7 @@
|
|
| 1 |
import streamlit as st
|
| 2 |
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
|
| 3 |
|
| 4 |
-
|
| 5 |
-
return pipeline("translation", model=model_name)
|
| 6 |
-
|
| 7 |
-
model_name = "Helsinki-NLP/opus-mt-en-ru"
|
| 8 |
-
translator = initialize_translator(model_name)
|
| 9 |
|
| 10 |
def translate_text(text):
|
| 11 |
if text:
|
|
@@ -14,26 +10,9 @@ def translate_text(text):
|
|
| 14 |
return ""
|
| 15 |
|
| 16 |
st.title("Text Translation App")
|
| 17 |
-
|
| 18 |
-
st.sidebar.header("Settings")
|
| 19 |
-
language_pair = st.sidebar.selectbox(
|
| 20 |
-
"Choose language pair:",
|
| 21 |
-
[
|
| 22 |
-
"English to Russian (Helsinki-NLP/opus-mt-en-ru)",
|
| 23 |
-
"Russian to English (Helsinki-NLP/opus-mt-ru-en)"
|
| 24 |
-
]
|
| 25 |
-
)
|
| 26 |
-
|
| 27 |
-
if "Russian to English" in language_pair:
|
| 28 |
-
model_name = "Helsinki-NLP/opus-mt-ru-en"
|
| 29 |
-
else:
|
| 30 |
-
model_name = "Helsinki-NLP/opus-mt-en-ru"
|
| 31 |
-
|
| 32 |
-
translator = initialize_translator(model_name)
|
| 33 |
-
|
| 34 |
st.subheader("Enter text to translate:")
|
| 35 |
|
| 36 |
-
user_input = st.text_area("Your text here
|
| 37 |
|
| 38 |
if st.button("Translate"):
|
| 39 |
translation = translate_text(user_input)
|
|
|
|
| 1 |
import streamlit as st
|
| 2 |
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
|
| 3 |
|
| 4 |
+
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
|
| 6 |
def translate_text(text):
|
| 7 |
if text:
|
|
|
|
| 10 |
return ""
|
| 11 |
|
| 12 |
st.title("Text Translation App")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
st.subheader("Enter text to translate:")
|
| 14 |
|
| 15 |
+
user_input = st.text_area("Your text here:", height=200)
|
| 16 |
|
| 17 |
if st.button("Translate"):
|
| 18 |
translation = translate_text(user_input)
|