seal345 commited on
Commit
ac109fd
1 Parent(s): f79ae2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ import tokenizers
7
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
8
 
9
 
 
10
  def load_model():
11
  model = AutoModelForSequenceClassification.from_pretrained('model_roberta_trained', use_auth_token=True)
12
  tokenizer = AutoTokenizer.from_pretrained(
 
7
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
8
 
9
 
10
+ @st.cache(suppress_st_warning=True, hash_funcs={tokenizers.Tokenizer: lambda _: None})
11
  def load_model():
12
  model = AutoModelForSequenceClassification.from_pretrained('model_roberta_trained', use_auth_token=True)
13
  tokenizer = AutoTokenizer.from_pretrained(