Pegasus Fine-tuned for Summarization (SAMSum)
This model is a fine-tuned version of google/pegasus-cnn_dailymail on the SAMSum dataset. It is designed to generate concise and accurate summaries of daily conversations.
Model Details
- Base Model: Pegasus (Fine-tuned)
- Dataset: SAMSum (Conversational Summarization)
- Task: Text Summarization
- Language: English
Usage
You can use this model directly with the Hugging Face pipeline:
from transformers import pipeline
pipe = pipeline("summarization", model="mhmdrza9789/pegasus-samsum-summarizer")
dialogue = """
Hannah: Hey, do you have any plans for tonight?
Mark: Not really. Just staying in and watching some Netflix. Why?
"""
print(pipe(dialogue)[0]['summary_text'])