Downloads · 30 days
10
0% of all-time downloads
summarizer-cnndm
yuvraj/summarizer-cnndm
summarizer-cnndm is a summarization model from yuvraj. Use it when you need a shorter version of a longer text. It is set up for transformers.
BartForConditionalGeneration model fine tuned for summarization on 10000 samples from the cnn-dailymail dataset PyTorch model available from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline…
All-time downloads
2.6K
Public
Repo size
1.1 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin558 MB · 100%
At a glance
- Task
- Summarization
- Library
- transformers
- Model type
- bart
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Dec 11, 2020
- SHA
- 4dfc2c9a
Try a prompt
Model card
From the Hugging Face model README
Summarization
Model description
BartForConditionalGeneration model fine tuned for summarization on 10000 samples from the cnn-dailymail dataset
How to use
PyTorch model available
from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline
tokenizer = AutoTokenizer.from_pretrained("yuvraj/summarizer-cnndm")
AutoModelWithLMHead.from_pretrained("yuvraj/summarizer-cnndm")
summarizer = pipeline('summarization', model=model, tokenizer=tokenizer)
summarizer("<Text to be summarized>")
## Limitations and bias
Trained on a small dataset
- Task
- Summarization
- Library
- transformers
- Type
- bart
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- Dec 11, 2020