thon | |
from transformers import T5Tokenizer, T5ForConditionalGeneration | |
tokenizer = T5Tokenizer.from_pretrained("google-t5/t5-small") | |
model = T5ForConditionalGeneration.from_pretrained("google-t5/t5-small") | |
input_ids = tokenizer("The walks in park", return_tensors="pt").input_ids | |
sequence_ids = model.generate(input_ids) | |
sequences = tokenizer.batch_decode(sequence_ids) | |
sequences | |
[' park offers the park.'] |