kairaamilanii's picture
Update README.md
f57fa72 verified
metadata
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)