translator / app.py
KateProxa's picture
Rename FP.py to app.py
6177df3
Raw History Blame Contribute Delete
614 Bytes
import io
import streamlit as st
from transformers import pipeline
@st.cache_resource
def load_model():
return pipeline("translation_en_to_de", model = "Helsinki-NLP/opus-mt-en-de")
translation = load_model()
st.title('Translator to Deutch')
st.header('Это приложение для перевода текста на немецкий язык')
text = st.text_input('Введите текст для перевода на английском языке')
result = st.button('Перевести')
if result:
tr_text = translation(text)
st.write("Перевод:", tr_text[0]['translation_text'])