T5 translation example

t5_translate.py

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("t5-base")
model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")

example_input = "This sentence is false"
input_ids = tokenizer("translate English to German: "+example_input, return_tensors="pt").input_ids
outputs = model.generate(input_ids)
decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)

print(decoded)