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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -7,7 +7,6 @@ import tokenizers
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(
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(