language: "en" | |
tags: | |
- summarization | |
β | |
# Summarization | |
β | |
## Model description | |
β | |
BartForConditionalGeneration model fine tuned for summarization on 10000 samples from the cnn-dailymail dataset | |
β | |
## How to use | |
β | |
PyTorch model available | |
β | |
```python | |
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 | |