---
license: unknown
datasets:
- fajrikoto/id_liputan6
language:
- id
metrics:
- rouge
base_model:
- cahya/bert2bert-indonesian-summarization
pipeline_tag: summarization

model-index:
  - name: BERT2BERT-Summarizer
    results:
      - task:
          type: text-summarization
        dataset:
          name: liputan6
          type: canonical
        metrics:
          - name: ROUGE-1
            type: ROUGE-1
            value: 0.484
      - task:
          type: text-summarization
        dataset:
          name: liputan6
          type: canonical
        metrics:
          - name: ROUGE-2
            type: ROUGE-2
            value: 0.399
      - task:
          type: text-summarization
        dataset:
          name: liputan6
          type: canonical
        metrics:
          - name: ROUGE-L
            type: ROUGE-L
            value: 0.447

---


model_name = "kairaamilanii/BERT2BERT-ExtractiveSummarizer"

tokenizer = BertTokenizer.from_pretrained(model_name)

tokenizer.bos_token = tokenizer.cls_token

tokenizer.eos_token = tokenizer.sep_token

model = EncoderDecoderModel.from_pretrained(model_name)