Downloads · 30 days
141
1% of all-time downloads
bart-base-samsum
lidiya/bart-base-samsum
bart-base-samsum is a summarization model from lidiya. Use it when you need a shorter version of a longer text. It is set up for transformers. The card lists the license as apache-2.0.
This model was obtained by fine-tuning facebook/bart-base on Samsum dataset.
All-time downloads
19.7K
Public
Parameters
139M
1.1 GB on disk
Likes
4
Public
Repo makeup
Click a slice to open those files.
.bin558 MB · 50%
At a glance
- Task
- Summarization
- Library
- transformers
- License
- apache-2.0
- Model type
- bart
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Mar 17, 2023
- SHA
- 6d9e0d7b
Try a prompt
Datasets
Model card
From the Hugging Face model README
bart-base-samsum
This model was obtained by fine-tuning facebook/bart-base on Samsum dataset.
Usage
from transformers import pipeline
summarizer = pipeline("summarization", model="lidiya/bart-base-samsum")
conversation = '''Jeff: Can I train a 🤗 Transformers model on Amazon SageMaker?
Philipp: Sure you can use the new Hugging Face Deep Learning Container.
Jeff: ok.
Jeff: and how can I get started?
Jeff: where can I find documentation?
Philipp: ok, ok you can find everything here. https://huggingface.co/blog/the-partnership-amazon-sagemaker-and-hugging-face
'''
summarizer(conversation)
Training procedure
- Colab notebook: https://colab.research.google.com/drive/1RInRjLLso9E2HG_xjA6j8JO3zXzSCBRF?usp=sharing
Results
| key | value |
|---|---|
| eval_rouge1 | 46.6619 |
| eval_rouge2 | 23.3285 |
| eval_rougeL | 39.4811 |
| eval_rougeLsum | 43.0482 |
| test_rouge1 | 44.9932 |
| test_rouge2 | 21.7286 |
| test_rougeL | 38.1921 |
| test_rougeLsum | 41.2672 |
- Task
- Summarization
- Library
- transformers
- Type
- bart
- License
- apache-2.0
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- Mar 17, 2023