Add application file

app.py

@@ -1,22 +1,12 @@
 import streamlit as st
-from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
+from transformers import pipeline
 
-# Загружаем модель и токенизатор
-model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
-tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
-
-# Функция перевода
-def translate_text(text, src_lang, tgt_lang):
-    tokenizer.src_lang = src_lang
-    encoded = tokenizer(text, return_tensors="pt")
-    generated_tokens = model.generate(
-        **encoded,
-        forced_bos_token_id=tokenizer.lang_code_to_id[tgt_lang]
-    )
-    return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
+# Модели для перевода
+ru_en_translator = pipeline('translation_ru_to_en', model='Helsinki-NLP/opus-mt-ru-en')
+en_ru_translator = pipeline('translation_en_to_ru', model='Helsinki-NLP/opus-mt-en-ru')
 
 # Название и описание
-st.title("Переводчик MBart")
+st.title("Переводчик")
 st.write("Перевод между русским и английским языками.")
 
 # Ввод текста и выбор языков
@@ -34,10 +24,17 @@ tgt_lang = st.selectbox("Выберите целевой язык:", list(langua
 # Перевод при нажатии кнопки
 if st.button("Перевести"):
     if tgt_lang == src_lang:
-        st.write("Выберите только другой язык")
+        st.write("Выберите другой язык для перевода")
     elif input_text.strip():
-        translation = translate_text(input_text, language_options[src_lang], language_options[tgt_lang])
-        st.write(f"Перевод с {src_lang} на {tgt_lang}:")
-        st.write(translation[0])
+        if src_lang == "Русский" and tgt_lang == "Английский":
+            # Перевод с русского на английский
+            translation = ru_en_translator(input_text)
+            st.write(f"Перевод с Русского на Английский:")
+            st.write(translation[0]['translation_text'])
+        elif src_lang == "Английский" and tgt_lang == "Русский":
+            # Перевод с английского на русский
+            translation = en_ru_translator(input_text)
+            st.write(f"Перевод с Английского на Русский:")
+            st.write(translation[0]['translation_text'])
     else:
         st.write("Пожалуйста, введите текст для перевода.")