kairaamilanii's picture
Update README.md
f57fa72 verified
---
license: unknown
datasets:
- fajrikoto/id_liputan6
language:
- id
metrics:
- rouge
base_model:
- cahya/t5-base-indonesian-summarization-cased
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.525
- task:
type: text-summarization
dataset:
name: liputan6
type: canonical
metrics:
- name: ROUGE-2
type: ROUGE-2
value: 0.448
- task:
type: text-summarization
dataset:
name: liputan6
type: canonical
metrics:
- name: ROUGE-L
type: ROUGE-L
value: 0.491
---
model_path = "kairaamilanii/T5-ExtractiveSummarizer"
summarizer = T5ForConditionalGeneration.from_pretrained(model_path)
tokenizer = T5Tokenizer.from_pretrained(model_path)