julien-c HF staff commited on
Commit
4dfc2c9
β€’
1 Parent(s): 1177968

Migrate model card from transformers-repo

Browse files

Read announcement at https://discuss.huggingface.co/t/announcement-all-model-cards-will-be-migrated-to-hf-co-model-repos/2755
Original file history: https://github.com/huggingface/transformers/commits/master/model_cards/yuvraj/summarizer-cnndm/README.md

Files changed (1) hide show
  1. README.md +27 -0
README.md ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: "en"
3
+ tags:
4
+ - summarization
5
+ ---
6
+ ​
7
+ # Summarization
8
+ ​
9
+ ## Model description
10
+ ​
11
+ BartForConditionalGeneration model fine tuned for summarization on 10000 samples from the cnn-dailymail dataset
12
+ ​
13
+ ## How to use
14
+ ​
15
+ PyTorch model available
16
+ ​
17
+ ```python
18
+ from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline
19
+ ​
20
+ tokenizer = AutoTokenizer.from_pretrained("yuvraj/summarizer-cnndm")
21
+ AutoModelWithLMHead.from_pretrained("yuvraj/summarizer-cnndm")
22
+ ​
23
+ summarizer = pipeline('summarization', model=model, tokenizer=tokenizer)
24
+ summarizer("<Text to be summarized>")
25
+ ​
26
+ ## Limitations and bias
27
+ Trained on a small dataset