Update README.md

README.md

@@ -91,11 +91,13 @@ encoding = tokenizer(text,question,add_special_tokens = True,
                      truncation = 'only_first',
                      max_length = 32)
 out = model(encoding['input_ids'].to(device),encoding['attention_mask'].to(device), encoding['token_type_ids'].to(device))
-start_index, end_index = generate_indexes(out['start_logits'][0][1:], out['end_logits'][0][1:], 5, 0)
-print(tokenizer.tokenize(text + question)[start_index:end_index+1])
+#we had to change some pieces of code to make it compatible 
+#with one answer generation at a time
+answer_start_index, answer_end_index = generate_indexes(out['start_logits'][0][1:], out['end_logits'][0][1:], 5, 0)
 print(tokenizer.tokenize(text + question))
->>> ['▁26']
+print(tokenizer.tokenize(text + question)[answer_start_index : (answer_end_index + 1)])
 >>> ['▁سلام', '▁من', '▁پدر', 'ام', 'م', '▁26', '▁سالم', 'ه', 'نام', 'م', '▁چیست', '؟']
+>>> ['▁26']
 ```
 
 ## Acknowledgments