Spaces:

vives
/

bert_cvent_top_k_sim

Runtime error

vives commited on Jun 8, 2022

Commit

4d020be

1 Parent(s): 8428c07

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import pickle
 import numpy as np
 import itertools
-@st.cache(hash_funcs={AutoTokenizer: lambda _: None, AutoModelForMaskedLM:lambda _: None})
 def load_bert():
   return (AutoModelForMaskedLM.from_pretrained("vives/distilbert-base-uncased-finetuned-cvent-2019_2022", output_hidden_states=True),
   AutoTokenizer.from_pretrained("vives/distilbert-base-uncased-finetuned-cvent-2019_2022"))

 import numpy as np
 import itertools
+@st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, AutoModelForMaskedLM:lambda _: None})
 def load_bert():
   return (AutoModelForMaskedLM.from_pretrained("vives/distilbert-base-uncased-finetuned-cvent-2019_2022", output_hidden_states=True),
   AutoTokenizer.from_pretrained("vives/distilbert-base-uncased-finetuned-cvent-2019_2022"))