facebook
/

genre-linking-aidayago2

Text2Text Generation

entity-retrieval

named-entity-disambiguation

entity-disambiguation

named-entity-linking

Inference Endpoints

Model card Files Files and versions Community

nicoladecao commited on Jun 7, 2022

Commit

2695201

•

1 Parent(s): bd735c2

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -46,8 +46,8 @@ import pickle
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 # 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))

 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 # OPTIONAL: load the prefix tree (trie), you need to additionally download
+# https://huggingface.co/facebook/genre-linking-aidayago2/blob/main/trie.py and
+# https://huggingface.co/facebook/genre-linking-aidayago2/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))