Update README.md

README.md

@@ -20,7 +20,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # xlm-roberta-base-finetuned-JaQuAD
 
-This model is a fine-tuned version of [xlm-roberta-base](https://huggingface.co/xlm-roberta-base) on the ja_qu_ad dataset.
+This model is a fine-tuned version of [xlm-roberta-base](https://huggingface.co/xlm-roberta-base) on the [JaQuAD](https://huggingface.co/datasets/SkelterLabsInc/JaQuAD) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.7495
 
@@ -33,6 +33,24 @@ More information needed
 ```python
 import torch
 from transformers import AutoModelForQuestionAnswering, AutoTokenizer
+
+text = "2015年9月1日、私は横浜へ車で出かけました。映画を観た後に中華街まで電車で行き、昼ご飯は重慶飯店で中華フルコースを食べました。"
+questions= ["どこへ出かけた？", "電車に乗る前は何をしていた？", "重慶飯店で何を食べた？", "いつ横浜に出かけた？"]
+
+for question in questions:
+    
+    inputs = tokenizer.encode_plus(question, text, add_special_tokens=True, return_tensors="pt")
+    
+    with torch.no_grad():
+        output = model(**inputs)
+
+    answer_start = torch.argmax(output.start_logits)
+    answer_end = torch.argmax(output.end_logits)
+
+    answer_tokens = inputs.input_ids[0, answer_start : answer_end + 1]
+    answer = tokenizer.decode(answer_tokens)
+
+    print(f"質問: {question} -> 回答: {answer}")
 ```
 
 ## Training and evaluation data