Update README.md

README.md

@@ -75,7 +75,7 @@ def generate_indexes(start_logits, end_logits, N, min_index_list):
   final_start_idx.append(start_idx)    
   final_end_idx.append(end_idx)      
 
-  return final_start_idx, final_end_idx
+  return final_start_idx[0], final_end_idx[0]
 ```
 
 ```python
@@ -91,9 +91,10 @@ 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))
-print(generate_indexes(out['start_logits'][0][1:], out['end_logits'][0][1:], 5, 0))
+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])
 print(tokenizer.tokenize(text + question))
->>> ([5], [5])
+>>> ['▁26']
 >>> ['▁سلام', '▁من', '▁پدر', 'ام', 'م', '▁26', '▁سالم', 'ه', 'نام', 'م', '▁چیست', '؟']
 ```