Update README.md

README.md

@@ -42,12 +42,12 @@ This model was trained on the full training set of [BLINK](https://arxiv.org/abs
 Here is an example of generation for Wikipedia page disambiguation:
 
 ```python
-import pickle
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 # OPTIONAL: load the prefix tree (trie), you need to additionally download
 # https://huggingface.co/facebook/genre-linking-blink/blob/main/trie.py and 
 # https://huggingface.co/facebook/genre-linking-blink/blob/main/kilt_titles_trie_dict.pkl
+# import pickle
 # from trie import Trie
 # with open("kilt_titles_trie_dict.pkl", "rb") as f:
 #     trie = Trie.load_from_dict(pickle.load(f))