pszemraj commited on
Commit
bc3ab92
1 Parent(s): 9043273

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -8
README.md CHANGED
@@ -82,8 +82,7 @@ inference:
82
  ## Other Checkpoints on Booksum
83
 
84
  - See [led-large-book-summary](https://huggingface.co/pszemraj/led-large-book-summary) for LED-large trained on the same dataset.
85
- - A version of bigbird-pegasus trained on booksum is [here](https://huggingface.co/pszemraj/bigbird-pegasus-large-K-booksum)
86
- - original baseline checkpoint [`pszemraj/led-base-16384-finetuned-booksum`](https://huggingface.co/pszemraj/led-base-16384-finetuned-booksum)
87
  ---
88
 
89
  # Usage - Basics
@@ -122,17 +121,16 @@ wall_of_text = "your words here"
122
 
123
  result = summarizer(
124
  wall_of_text,
125
- min_length=16,
126
  max_length=256,
127
  no_repeat_ngram_size=3,
128
- encoder_no_repeat_ngram_size =3,
129
- clean_up_tokenization_spaces=True,
130
- repetition_penalty=3.7,
131
  num_beams=4,
 
132
  early_stopping=True,
133
  )
134
-
135
-
136
  ```
137
 
138
  ---
 
82
  ## Other Checkpoints on Booksum
83
 
84
  - See [led-large-book-summary](https://huggingface.co/pszemraj/led-large-book-summary) for LED-large trained on the same dataset.
85
+
 
86
  ---
87
 
88
  # Usage - Basics
 
121
 
122
  result = summarizer(
123
  wall_of_text,
124
+ min_length=8,
125
  max_length=256,
126
  no_repeat_ngram_size=3,
127
+ encoder_no_repeat_ngram_size=3,
128
+ repetition_penalty=3.5,
 
129
  num_beams=4,
130
+ do_sample=False,
131
  early_stopping=True,
132
  )
133
+ print(result[0]['generated_text'])
 
134
  ```
135
 
136
  ---