Translate_app / app.py
KateProxa's picture
update
e75f82c
import io
import streamlit as st
from transformers import pipeline
@st.cache_resource
#@st.cache_data
# загружаю модель
def load_model():
return pipeline("translation_ru_to_fr", model = "Helsinki-NLP/opus-mt-ru-fr")
translation = load_model()
# Вывожу заголовок страницы средствами Streamlit
st.title('Translator from Russian to French')
st.write('Это приложение для перевода текста с русского языка на французский')
text = st.text_area('Введите текст для перевода', 'Скоро наступит Новый Год!')
# Кнопка для запуска
result = st.button('Перевести')
# Если кнопка нажата, то запускаем
if result:
tr_text = translation(text)
st.write("Перевод:", tr_text[0]['translation_text'])