Views
No views yet
1
2from transformers import PegasusForConditionalGeneration, PegasusTokenizer
3
4# Load the pre-trained model and tokenizer
5model_name = "ailm/pegsus-text-summarization"
6model = PegasusForConditionalGeneration.from_pretrained(model_name)
7tokenizer = PegasusTokenizer.from_pretrained(model_name)
8
9# Define the input text
10text = "Your input text here"
11
12# Tokenize the input text
13tokens = tokenizer(text, truncation=True, padding="longest", return_tensors="pt")
14
15# Generate the summary
16summary = model.generate(**tokens)
17
18# Decode and print the summary
19print(tokenizer.decode(summary[0], skip_special_tokens=True))
20