Update README.md

README.md

@@ -177,14 +177,21 @@ sparse_rep = expander.expand(
 ```
 
  
+
 ## 8. With HuggingFace
 
+**NOTEBOOK USERS, please login first**
+```
+!huggingface-cli login
+```
+
 ```python
 import torch
 from transformers import AutoModelForMaskedLM, AutoTokenizer
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 tokenizer = AutoTokenizer.from_pretrained('prithivida/Splade_PP_en_v1')
+reverse_voc = {v: k for k, v in tokenizer.vocab.items()}
 model = AutoModelForMaskedLM.from_pretrained('prithivida/Splade_PP_en_v1')
 model.to(device)
 
@@ -216,6 +223,7 @@ for k, v in sorted_d.items():
     bow_rep.append((reverse_voc[k], round(v,2)))
 
 print("SPLADE BOW rep:\n", bow_rep)
+
 ```
 
 ## BEIR Zeroshot ODD performance: