ilyshi commited on
Commit
399f3ee
1 Parent(s): 51ce1a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -31,7 +31,6 @@ def make_prediction(text):
31
 
32
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
33
 
34
- @st.cache
35
  model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
36
  model_name = "trained_model2"
37
  model_path = model_name + '.zip'
 
31
 
32
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
33
 
 
34
  model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
35
  model_name = "trained_model2"
36
  model_path = model_name + '.zip'