kairaamilanii commited on
Commit
51090b1
·
verified ·
1 Parent(s): e110fef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -0
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)