| 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.'] |