AdamCodd commited on
Commit
367e77e
1 Parent(s): 42dd1ff

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -48,7 +48,7 @@ More information needed
48
 
49
  ```python
50
  from transformers import pipeline
51
- summarizer = pipeline("summarization", model="AdamCodd/bart-large-cnn-samsum")
52
 
53
  conversation = '''Sugi: I am tired of everything in my life.
54
  Tommy: What? How happy your life is! I do envy you.
@@ -87,6 +87,6 @@ The following hyperparameters were used during training:
87
 
88
  ### Framework versions
89
 
90
- - Transformers 4.33.2
91
  - Pytorch lightning 2.0.9
92
  - Tokenizers 0.13.3
 
48
 
49
  ```python
50
  from transformers import pipeline
51
+ summarizer = pipeline("text2text-generation", model="AdamCodd/bart-large-cnn-samsum")
52
 
53
  conversation = '''Sugi: I am tired of everything in my life.
54
  Tommy: What? How happy your life is! I do envy you.
 
87
 
88
  ### Framework versions
89
 
90
+ - Transformers 4.34.0
91
  - Pytorch lightning 2.0.9
92
  - Tokenizers 0.13.3