nicoladecao commited on
Commit
3a55a60
1 Parent(s): 004dbaa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -42,12 +42,12 @@ This model was trained on the full training set of [BLINK](https://arxiv.org/abs
42
  Here is an example of generation for Wikipedia page disambiguation:
43
 
44
  ```python
45
- import pickle
46
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
47
 
48
  # OPTIONAL: load the prefix tree (trie), you need to additionally download
49
  # https://huggingface.co/facebook/genre-linking-blink/blob/main/trie.py and
50
  # https://huggingface.co/facebook/genre-linking-blink/blob/main/kilt_titles_trie_dict.pkl
 
51
  # from trie import Trie
52
  # with open("kilt_titles_trie_dict.pkl", "rb") as f:
53
  # trie = Trie.load_from_dict(pickle.load(f))
 
42
  Here is an example of generation for Wikipedia page disambiguation:
43
 
44
  ```python
 
45
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
46
 
47
  # OPTIONAL: load the prefix tree (trie), you need to additionally download
48
  # https://huggingface.co/facebook/genre-linking-blink/blob/main/trie.py and
49
  # https://huggingface.co/facebook/genre-linking-blink/blob/main/kilt_titles_trie_dict.pkl
50
+ # import pickle
51
  # from trie import Trie
52
  # with open("kilt_titles_trie_dict.pkl", "rb") as f:
53
  # trie = Trie.load_from_dict(pickle.load(f))