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)
|