update usage in readme.md

README.md

@@ -23,4 +23,29 @@ Training Arguments:
 
 - epoch: 3
 
-[Colab for detailed](https://colab.research.google.com/drive/1kZv7ZRmvUdCKEhQg8MBrKljGWvV2X3CP?usp=sharing)
+[Colab for detailed](https://colab.research.google.com/drive/1kZv7ZRmvUdCKEhQg8MBrKljGWvV2X3CP?usp=sharing)
+
+
+## Usage
+
+### In Transformers
+
+```python
+text = "馬雲是我爸爸。"
+query = "我爸爸是誰？"
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+tokenizer = BertTokenizerFast.from_pretrained("bert_drcd_384")
+model = BertForQuestionAnswering.from_pretrained("bert_drcd_384").to(device)
+encoded_input = tokenizer(text, query, return_tensors="pt").to(device)
+qa_outputs = model(**encoded_input)
+
+start = torch.argmax(qa_outputs.start_logits).item()
+end = torch.argmax(qa_outputs.end_logits).item()
+answer = encoded_input.input_ids.tolist()[0][start : end + 1]
+answer = "".join(tokenizer.decode(answer).split())
+
+start_prob = torch.max(torch.nn.Softmax(dim=-1)(qa_outputs.start_logits)).item()
+end_prob = torch.max(torch.nn.Softmax(dim=-1)(qa_outputs.end_logits)).item()
+confidence = (start_prob + end_prob) / 2
+print(answer, confidence) # 馬雲, 0.98
+```