T5-Abstractive / README.md
migz117's picture
Update README.md
e82dfe8 verified
metadata
datasets:
  - fajrikoto/id_liputan6
language:
  - id
pipeline_tag: summarization
tags:
  - summarization
  - t5
metrics:
  - bertscore
  - rouge

Indonesian T5 Abstractive Summarization Base Model

Hello everyone, we are from Bina Nusantara University (SumText Group) consisting of Stevan Pohan, Joseph Vincent Liem, and Yongky Alexander Tristan. This is the result of a model that we have fine-tuned for the use of abstractive summarization

Load Fine Tuned Model

  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
  model_path = "migz117/T5-Abstractive"
  model = AutoModelForSeq2SeqLM.from_pretrained(model_path).to(device)
  tokenizer = AutoTokenizer.from_pretrained(model_path)