TraducaoPT-EN / pages /traducao.py
thlanza's picture
Update pages/traducao.py
8e8c8ba
raw
history blame contribute delete
808 Bytes
import streamlit as st
from langchain.llms import HuggingFaceHub
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
#Function to return the response
def load_answer(question):
tokenizer = AutoTokenizer.from_pretrained("unicamp-dl/translation-pt-en-t5")
model = AutoModelForSeq2SeqLM.from_pretrained("unicamp-dl/translation-pt-en-t5")
pten_pipeline = pipeline('text2text-generation', model=model, tokenizer=tokenizer)
return pten_pipeline(question)[0]["generated_text"]
#Gets the user input
def get_text():
input_text = st.text_input("Sua frase em português: ", key="input")
return input_text
user_input=get_text()
response = load_answer(user_input)
submit = st.button('Traduzir para inglês')
if submit:
st.subheader("Answer:")
st.write(response)