Commit
·
7be9dc0
1
Parent(s):
78c238a
Update README.md
Browse files
README.md
CHANGED
|
@@ -31,7 +31,7 @@ Goal: a model that can be used for a general-purpose summarizer for academic and
|
|
| 31 |
|
| 32 |
## Usage
|
| 33 |
|
| 34 |
-
Check the colab notebook. **The model expects a prompt prepended to the source document to indicate the type of summary**, examples of prompts used to train the model here:
|
| 35 |
```
|
| 36 |
prompts = {
|
| 37 |
"article": "Produce an article summary of the following news article:",
|
|
@@ -56,7 +56,6 @@ results = summarizer(
|
|
| 56 |
num_beams=5,
|
| 57 |
min_length=5,
|
| 58 |
no_repeat_ngram_size=3,
|
| 59 |
-
skip_special_tokens=True,
|
| 60 |
truncation=True,
|
| 61 |
max_length=512,
|
| 62 |
)
|
|
|
|
| 31 |
|
| 32 |
## Usage
|
| 33 |
|
| 34 |
+
Check the colab notebook for desired usage. **The model expects a prompt prepended to the source document to indicate the type of summary**, examples of prompts used to train the model here:
|
| 35 |
```
|
| 36 |
prompts = {
|
| 37 |
"article": "Produce an article summary of the following news article:",
|
|
|
|
| 56 |
num_beams=5,
|
| 57 |
min_length=5,
|
| 58 |
no_repeat_ngram_size=3,
|
|
|
|
| 59 |
truncation=True,
|
| 60 |
max_length=512,
|
| 61 |
)
|