kairaamilanii's picture
Update README.md
dd1e9ee verified
metadata
license: unknown
datasets:
  - fajrikoto/id_liputan6
language:
  - id
metrics:
  - rouge
base_model:
  - cahya/t5-base-indonesian-summarization-cased
model-index:
  - name: BERT2BERT-Summarizer
    results:
      - task:
          type: text-summarization
        dataset:
          name: liputan6
          type: canonical
        metrics:
          - name: BERTScore-Precision
            type: Precision
            value: 0.727
      - task:
          type: text-summarization
        dataset:
          name: liputan6
          type: canonical
        metrics:
          - name: BERTScore-Recall
            type: Recall
            value: 0.741
      - task:
          type: text-summarization
        dataset:
          name: liputan6
          type: canonical
        metrics:
          - name: BERTScore-F1
            type: F1
            value: 0.734

model_path = "kairaamilanii/T5-AbstractiveSummarizer"

summarizer = T5ForConditionalGeneration.from_pretrained(model_path)

tokenizer = T5Tokenizer.from_pretrained(model_path)