Update README.md
Browse files
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 |
-
|
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=
|
126 |
max_length=256,
|
127 |
no_repeat_ngram_size=3,
|
128 |
-
encoder_no_repeat_ngram_size
|
129 |
-
|
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 |
---
|