Update README.md

README.md

@@ -76,18 +76,13 @@ def generate_indexes(start_logits, end_logits, N, min_index_list):
   final_end_idx.append(end_idx)      
 
   return final_start_idx, final_end_idx
-
-print(generate_indexes(out['start_logits'][0][1:], out['end_logits'][0][1:], 5, 0))
-print(tokenizer.tokenize(text+question))
 ```
 
 ```python
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 model.eval().to(device)
-
 text = 'سلام من پدرامم 26 سالمه'
 question = 'نامم چیست؟'
-print(tokenizer.tokenize(text + question))
 encoding = tokenizer(text,question,add_special_tokens = True,
                      return_token_type_ids = True,
                      return_tensors = 'pt',
@@ -96,7 +91,9 @@ 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))
+print(tokenizer.tokenize(text + question))
+>>> ([5], [5])
 >>> ['▁سلام', '▁من', '▁پدر', 'ام', 'م', '▁26', '▁سالم', 'ه', 'نام', 'م', '▁چیست', '؟']
 ```