AdamCodd commited on
Commit
412754b
1 Parent(s): c4e82ec

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -27,6 +27,7 @@ model-index:
27
  - type: rouge-l
28
  value: 41.214
29
  name: Validation ROUGE-L
 
30
  ---
31
 
32
  # bart-large-cnn-samsum
@@ -47,7 +48,7 @@ More information needed
47
 
48
  ```python
49
  from transformers import pipeline
50
- summarizer = pipeline("text2text-generation", model="AdamCodd/bart-large-cnn-samsum")
51
 
52
  conversation = '''Emily: Hey Alex, have you heard about the new restaurant that opened downtown?
53
  Alex: No, I haven't. What's it called?
 
27
  - type: rouge-l
28
  value: 41.214
29
  name: Validation ROUGE-L
30
+ pipeline_tag: summarization
31
  ---
32
 
33
  # bart-large-cnn-samsum
 
48
 
49
  ```python
50
  from transformers import pipeline
51
+ summarizer = pipeline("summarization", model="AdamCodd/bart-large-cnn-samsum")
52
 
53
  conversation = '''Emily: Hey Alex, have you heard about the new restaurant that opened downtown?
54
  Alex: No, I haven't. What's it called?