wuriyanto commited on
Commit
9260330
·
verified ·
1 Parent(s): 2bf4a5b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +30 -3
README.md CHANGED
@@ -1,3 +1,30 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ ---
4
+ # Indonesian BERT Question Answer Uncased
5
+
6
+ ### Model Description
7
+
8
+ This model uses the `bert-base-multilingual-uncased` basic model and is trained for the question-answer task using Google's TyDi QA dataset.
9
+
10
+ ### Usage
11
+ ```python
12
+ from transformers import AutoTokenizer, BertForQuestionAnswering
13
+ import torch
14
+
15
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_OUTPUT_BEST_NAME)
16
+ model = BertForQuestionAnswering.from_pretrained(MODEL_OUTPUT_BEST_NAME)
17
+
18
+ question = "apa nama lahir soekarno?"
19
+ text = "Dr.(HC) Ir. Soekarno (ER, EYD: Sukarno, nama lahir: Koesno Sosrodihardjo) (lahir di Surabaya, Jawa Timur, 6 Juni 1901 - meninggal di Jakarta, 21 Juni 1970 pada umur 69 tahun) adalah Presiden Indonesia pertama yang menjabat pada periode 1945-1966.Setelah pertanggung jawabannya ditolak Majelis Permusyawaratan Rakyat Sementara (MPRS) pada sidang umum ke empat tahun 1967, Soekarno diberhentikan dari jabatannya sebagai presiden pada Sidang Istimewa MPRS pada tahun yang sama dan Soeharto menggantikannya sebagai pejabat Presiden Republik Indonesia."
20
+
21
+ inputs = tokenizer(question, text, return_tensors="pt")
22
+ with torch.no_grad():
23
+ outputs = model(**inputs)
24
+
25
+ answer_start_index = outputs.start_logits.argmax()
26
+ answer_end_index = outputs.end_logits.argmax()
27
+
28
+ predict_answer_tokens = inputs.input_ids[0, answer_start_index : answer_end_index + 1]
29
+ tokenizer.decode(predict_answer_tokens, skip_special_tokens=True)
30
+ ```