Update README.md

README.md

@@ -25,11 +25,18 @@ Here is how to use this model to get the features of a given text in PyTorch:
 from transformers import AutoModelForQuestionAnswering, AutoTokenizer
 
 model = AutoModelForQuestionAnswering.from_pretrained("rubentito/bert-large-mpdocvqa")
+tokenizer = AutoTokenizer.from_pretrained("rubentito/bert-large-mpdocvqa")
 
 question = "Replace me by any text you'd like."
 context = "Put some context for answering"
+
 encoded_input = tokenizer(question, context, return_tensors='pt')
 output = model(**encoded_input)
+
+start_pos = output.start_logits.argmax(dim=-1).item()
+end_pos = output.end_logits.argmax(dim=-1).item()
+
+pred_answer = context[start_pos:end_pos]
 ```
 
 ## Model results