idt5-qa-qg / README.md
muchad's picture
Update README.md
84d08a3
|
raw
history blame
2.24 kB
metadata
language: id
datasets:
  - SQuADv2.0
tags:
  - question-generation
  - multitask-model
  - idt5
widget:
  - text: >-
      generate question: <hl> Dua orang <hl> pengembara berjalan di sepanjang
      jalan yang berdebu dan tandus di hari yang sangat panas. Tidak lama
      kemudian, mereka menemukan sebuah pohon besar. </s>
  - text: >-
      question: Siapa pemimpin Kerajaan Tarumanegara? context: Raja Purnawarman
      mulai memerintah Kerajaan Tarumanegara pada tahun 395 M. </s>
license: mit

idT5 for Indonesian Question Generation and Question Answering

idT5 (Indonesian version of mT5) is fine-tuned on 30% of translated SQuAD v2.0 for Question Generation and Question Answering tasks.

Live Demo

Requirements

!pip install transformers==4.4.2
!pip install sentencepiece==0.1.95
!git clone https://github.com/muchad/qaqg.git
%cd qaqg/

Usage πŸš€

Question Generation

Open In Colab

from pipeline_qg import pipeline #pipeline_qg.py script in the cloned repo
qg = pipeline()

#sample
qg("Raja Purnawarman mulai memerintah Kerajaan Tarumanegara pada tahun 395 M.")

#output
=> [{'answer': 'Raja Purnawarman','question': 'Siapa yang memerintah Kerajaan Tarumanegara?'}, {'answer': '395 M','question': 'Kapan Raja Purnawarman memerintah Kerajaan Tarumanegara?'}]

Question Answering

Open In Colab

from pipeline_qa import pipeline #pipeline_qa.py script in the cloned repo
qa = pipeline()

#sample
qa({"context":"Raja Purnawarman mulai memerintah Kerajaan Tarumanegara pada tahun 395 M.","question":"Siapa pemimpin Kerajaan Tarumanegara?"})

#output
=> Raja Purnawarman