Update README.md

README.md

@@ -47,8 +47,9 @@ Here is an example of generation for Wikipedia page retrieval for open-domain fa
 import pickle
 from transformers import BartTokenizer, BartForConditionalGeneration
 
-# OPTIONAL: load the prefix tree (trie), you need to download trie.py from
-# https://huggingface.co/facebook/genre-kilt/blob/main/trie.py
+# OPTIONAL: load the prefix tree (trie), you need to additionally download
+# https://huggingface.co/facebook/genre-kilt/blob/main/trie.py and 
+# https://huggingface.co/facebook/genre-kilt/blob/main/kilt_titles_trie_dict.pkl
 # from trie import Trie
 # with open("kilt_titles_trie_dict.pkl", "rb") as f:
 #     trie = Trie.load_from_dict(pickle.load(f))