rrevoid commited on
Commit
e739fa2
·
1 Parent(s): 6c23168

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import torch.nn as nn
6
  from transformers import RobertaTokenizer, RobertaModel
7
 
8
 
9
- @st.cache(hash_funcs={tokenizers.AddedToken: lambda _: None})
10
  def init():
11
  tokenizer = RobertaTokenizer.from_pretrained("roberta-large-mnli")
12
 
 
6
  from transformers import RobertaTokenizer, RobertaModel
7
 
8
 
9
+ @st.cache(hash_funcs={tokenizers.AddedToken: lambda _: None, _regex.Pattern: lambda _: None})
10
  def init():
11
  tokenizer = RobertaTokenizer.from_pretrained("roberta-large-mnli")
12