Update README.md
Browse files
README.md
CHANGED
@@ -27,6 +27,7 @@ Based on [BART Large](https://huggingface.co/transformers/model_doc/bart.html) w
|
|
27 |
Format your data so that each new article or evidence to add have `<EV>` token in front with each title prefixed by `<t>` and each abstract prefixed by `<abs>`. Please have the original summary also in the same format. You can have the list of articles and original summary concatenated in any order as long as they have the correct separator tokens.
|
28 |
|
29 |
```python
|
|
|
30 |
from transformers import LEDTokenizer, LEDForConditionalGeneration
|
31 |
|
32 |
tokenizer = LEDTokenizer.from_pretrained("hyesunyun/update-summarization-bart-large-longformer")
|
|
|
27 |
Format your data so that each new article or evidence to add have `<EV>` token in front with each title prefixed by `<t>` and each abstract prefixed by `<abs>`. Please have the original summary also in the same format. You can have the list of articles and original summary concatenated in any order as long as they have the correct separator tokens.
|
28 |
|
29 |
```python
|
30 |
+
import torch
|
31 |
from transformers import LEDTokenizer, LEDForConditionalGeneration
|
32 |
|
33 |
tokenizer = LEDTokenizer.from_pretrained("hyesunyun/update-summarization-bart-large-longformer")
|