bto00 commited on
Commit
4ee5fdf
·
1 Parent(s): 5bee2e1

Add application file

Browse files
Files changed (1) hide show
  1. app.py +34 -32
app.py CHANGED
@@ -1,40 +1,42 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- # Модели для перевода
5
- ru_en_translator = pipeline('translation_ru_to_en', model='Helsinki-NLP/opus-mt-ru-en')
6
- en_ru_translator = pipeline('translation_en_to_ru', model='Helsinki-NLP/opus-mt-en-ru')
 
 
7
 
8
- # Название и описание
9
- st.title("Переводчик")
10
- st.write("Перевод между русским и английским языками.")
11
 
12
- # Ввод текста и выбор языков
13
- input_text = st.text_area("Введите текст для перевода:")
14
 
15
- language_options = {
16
- "Русский": "ru_RU",
17
- "Английский": "en_XX"
18
- }
19
 
20
- # Пользователь выбирает исходный и целевой язык
21
- src_lang = st.selectbox("Выберите исходный язык:", list(language_options.keys()))
22
- tgt_lang = st.selectbox("Выберите целевой язык:", list(language_options.keys()))
23
 
24
- # Перевод при нажатии кнопки
25
- if st.button("Перевести"):
26
- if tgt_lang == src_lang:
27
- st.write("Выберите другой язык для перевода")
28
- elif input_text.strip():
29
- if src_lang == "Русский" and tgt_lang == "Английский":
30
- # Перевод с русского на английский
31
- translation = ru_en_translator(input_text)
32
- st.write(f"Перевод с Русского на Английский:")
33
- st.write(translation[0]['translation_text'])
34
- elif src_lang == "Английский" and tgt_lang == "Русский":
35
- # Перевод с английского на русский
36
- translation = en_ru_translator(input_text)
37
- st.write(f"Перевод с Английского на Русский:")
38
- st.write(translation[0]['translation_text'])
39
- else:
40
- st.write("Пожалуйста, введите текст для перевода.")
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ # Использование if __name__ == '__main__' для Windows
5
+ if __name__ == '__main__':
6
+ # Модели для перевода
7
+ ru_en_translator = pipeline('translation_ru_to_en', model='Helsinki-NLP/opus-mt-ru-en')
8
+ en_ru_translator = pipeline('translation_en_to_ru', model='Helsinki-NLP/opus-mt-en-ru')
9
 
10
+ # Название и описание
11
+ st.title("Переводчик")
12
+ st.write("Перевод между русским и английским языками.")
13
 
14
+ # Ввод текста и выбор языков
15
+ input_text = st.text_area("Введите текст для перевода:")
16
 
17
+ language_options = {
18
+ "Русский": "ru_RU",
19
+ "Английский": "en_XX"
20
+ }
21
 
22
+ # Пользователь выбирает исходный и целевой язык
23
+ src_lang = st.selectbox("Выберите исходный язык:", list(language_options.keys()))
24
+ tgt_lang = st.selectbox("Выберите целевой язык:", list(language_options.keys()))
25
 
26
+ # Перевод при нажатии кнопки
27
+ if st.button("Перевести"):
28
+ if tgt_lang == src_lang:
29
+ st.write("Выберите другой язык для перевода")
30
+ elif input_text.strip():
31
+ if src_lang == "Русский" and tgt_lang == "Английский":
32
+ # Перевод с русского на английский
33
+ translation = ru_en_translator(input_text)
34
+ st.write(f"Перевод с Русского на Английский:")
35
+ st.write(translation[0]['translation_text'])
36
+ elif src_lang == "Английский" and tgt_lang == "Русский":
37
+ # Перевод с английского на русский
38
+ translation = en_ru_translator(input_text)
39
+ st.write(f"Перевод с Английского на Русский:")
40
+ st.write(translation[0]['translation_text'])
41
+ else:
42
+ st.write("Пожалуйста, введите текст для перевода.")