kairaamilanii's picture
Update README.md
5c5ae5b verified
|
raw
history blame
405 Bytes
metadata
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)