Update README.md

README.md

@@ -6,6 +6,8 @@ datasets:
 model-index:
 - name: extractive_reader_nq_squad_v2
   results: []
+language:
+- en
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -50,3 +52,28 @@ The following hyperparameters were used during training:
 - Pytorch 1.13.1+cu117
 - Datasets 2.8.0
 - Tokenizers 0.13.2
+
+
+### Code Examples
+
+```python
+import torch
+import numpy as np
+from transformers import AutoTokenizer, AutoModelForQuestionAnswering
+
+tokenizer = AutoTokenizer.from_pretrained("ToluClassics/extractive_reader_nq_squad_v2")
+
+model = AutoModelForQuestionAnswering.from_pretrained("ToluClassics/extractive_reader_nq_squad_v2")
+
+question = ""
+context = ""
+
+inputs = tokenizer.encode(question, context, add_special_tokens=True, return_tensors="pt")
+
+output = model(inputs)
+
+answer_start = torch.argmax(output.start_logits)
+answer_end = torch.argmax(output.end_logits)
+if answer_end >= answer_start:
+    print(tokenizer.decode(inputs[0][answer_start:answer_end+1]))
+```