nicoladecao commited on
Commit
b683d97
1 Parent(s): 69836ea

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -174,7 +174,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
174
  # with open("titles_lang_all105_marisa_trie_with_redirect.pkl", "rb") as f:
175
  # trie = pickle.load(f)
176
 
177
- tokenizer = XLMRobertaTokenizer.from_pretrained("facebook/mgenre-wiki")
178
  model = AutoModelForSeq2SeqLM.from_pretrained("facebook/mgenre-wiki").eval()
179
 
180
  sentences = ["[START] Einstein [END] era un fisico tedesco."]
 
174
  # with open("titles_lang_all105_marisa_trie_with_redirect.pkl", "rb") as f:
175
  # trie = pickle.load(f)
176
 
177
+ tokenizer = AutoTokenizer.from_pretrained("facebook/mgenre-wiki")
178
  model = AutoModelForSeq2SeqLM.from_pretrained("facebook/mgenre-wiki").eval()
179
 
180
  sentences = ["[START] Einstein [END] era un fisico tedesco."]