riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-29 17:22:25 +06:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity
49e9be0639
transformers/model_cards/yuvraj/summarizer-cnndm/README.md
BatJedi abf5c56e9d
Created model card for my summarization model (#5838) 
* Created model card for my summarization model

* Update model_cards/yuvraj/summarizer-cnndm/README.md

Co-authored-by: Julien Chaumond <chaumond@gmail.com>
2020-07-21 03:54:14 -04:00

643 B
Raw Blame History

language tags
en
summarization

​

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