Update README.md
Browse files
README.md
CHANGED
@@ -10,3 +10,9 @@ base_model:
|
|
10 |
- cahya/bert2bert-indonesian-summarization
|
11 |
pipeline_tag: summarization
|
12 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
- cahya/bert2bert-indonesian-summarization
|
11 |
pipeline_tag: summarization
|
12 |
---
|
13 |
+
|
14 |
+
model_name = "kairaamilanii/BERT2BERT-Summarizer"
|
15 |
+
tokenizer = BertTokenizer.from_pretrained(model_name)
|
16 |
+
tokenizer.bos_token = tokenizer.cls_token
|
17 |
+
tokenizer.eos_token = tokenizer.sep_token
|
18 |
+
model = EncoderDecoderModel.from_pretrained(model_name).to(device)
|