translation / app.py
barrosm's picture
Update app.py
ec45e2c
raw
history blame contribute delete
547 Bytes
# translator api
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
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)
#pten_pipeline("translate Portuguese to English: Eu gosto de comer arroz.")
pten_pipeline("translate Portuguese to English: A bolsa de valores continua em alta apesar da grande volatilidade da cotação do dólar")