Update README.md

README.md

@@ -29,7 +29,7 @@ tokenizer = AutoTokenizer.from_pretrained('fkrasnov2/SBE')
 input_ids = tokenizer.encode("чёрное платье", max_length=model.config.max_position_embeddings, truncation=True, return_tensors='pt')
 
 model.eval()
-vector = model(input_ids=input_ids, attention_mask=input_ids>tokenizer.mask_token_id)[0][0,0]
+vector = model(input_ids=input_ids, attention_mask=input_ids!=tokenizer.pad_token_id)[0][0,0]
 
 assert model.config.hidden_size == vector.shape[0]
 ```