Sabina Muratova commited on
Commit
d9c64f9
·
1 Parent(s): fd459ff

Add application file

Browse files
Files changed (1) hide show
  1. app.py +2 -23
app.py CHANGED
@@ -1,11 +1,7 @@
1
  import streamlit as st
2
  from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
- def initialize_translator(model_name):
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 (e.g., 'The weather is nice today.'):", height=200)
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)