d.tsimerman commited on
Commit
f5e42ed
1 Parent(s): c7ca481
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ model_name = st.selectbox(
12
  )
13
  auth_token = os.environ.get('TOKEN') or True
14
 
15
- @st.cache(hash_funcs={tokenizers.Tokenizer: lambda tokenizer: hash(tokenizer.name_or_path)})
16
  def load_model(model_name: str):
17
  with st.spinner('Loading models...'):
18
  tokenizer = transformers.AutoTokenizer.from_pretrained(model_name, use_auth_token=auth_token)
12
  )
13
  auth_token = os.environ.get('TOKEN') or True
14
 
15
+ @st.cache(hash_funcs={tokenizers.Tokenizer: lambda tokenizer: hash(tokenizer.to_str())})
16
  def load_model(model_name: str):
17
  with st.spinner('Loading models...'):
18
  tokenizer = transformers.AutoTokenizer.from_pretrained(model_name, use_auth_token=auth_token)