Views
No views yet
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("figurative-nlp/t5-figurative-paraphrase")
model = AutoModelForSeq2SeqLM.from_pretrained("figurative-nlp/t5-figurative-paraphrase")
input_ids = tokenizer(
"paraphrase the sentence : i will talk this story to you from A to Z", return_tensors="pt"
).input_ids # Batch size 1
outputs = model.generate(input_ids,num_beams = 5)
result = tokenizer.decode(outputs[0], skip_special_tokens=True)
#result : i will talk this story to you from beginning to end..