Spaces:
Runtime error
Runtime error
File size: 1,449 Bytes
ec04958 52d6a93 b5fa92a 52d6a93 e4bc5ce 9d96326 ec04958 e4bc5ce ec04958 e4bc5ce ec04958 52d6a93 ec04958 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
import streamlit as st
from langchain.llms import HuggingFaceHub
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
# Answer:
# [
# 0:{
# "generated_text":"I like rice I like rice I like rice I like ri"
# }
# ]
#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"]
#App UI starts here
st.set_page_config(page_title="Tradução do Português para o Inglês.", page_icon=":robot:")
with st.sidebar:
st.sidebar.success("Selecione uma das seguintes opções")
st.image("https://www.viajenaviagem.com/wp-content/uploads/2020/02/belo-horizonte-pampulha.jpg.webp", caption='Autoria de Thiago Lanza. Todos os direitos reservados')
st.header("Tradução do Português para o inglês.")
st.subheader("Digite uma frase em português para tradução. Para melhores resultados, coloque um ponto final na frase.")
#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)
|