mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-30 01:32:23 +06:00
abf5c56e9d
* Created model card for my summarization model * Update model_cards/yuvraj/summarizer-cnndm/README.md Co-authored-by: Julien Chaumond <chaumond@gmail.com>
28 lines
643 B
Markdown
28 lines
643 B
Markdown
---
|
||
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
|