kairaamilanii's picture
Update README.md
5c5ae5b verified
|
raw
history blame
405 Bytes
---
license: unknown
datasets:
- fajrikoto/id_liputan6
language:
- id
metrics:
- rouge
base_model:
- cahya/bert2bert-indonesian-summarization
---
model_name = "kairaamilanii/BERT2BERT-AbstractiveSummarizer"
tokenizer = BertTokenizer.from_pretrained(model_name)
tokenizer.bos_token = tokenizer.cls_token
tokenizer.eos_token = tokenizer.sep_token
model = EncoderDecoderModel.from_pretrained(model_name)