Dzhamb commited on
Commit
4261a4c
·
1 Parent(s): c5becb1

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -32,7 +32,7 @@ def get_labels(text, model, tokenizer, count_labels=8):
32
  return result_labels
33
  result_labels.append(pair[1])
34
 
35
- @st.cache(allow_output_mutation=True)
36
  def load_model():
37
  tokenizer = DistilBertTokenizerFast()
38
  model = DistilBertForSequenceClassification()
 
32
  return result_labels
33
  result_labels.append(pair[1])
34
 
35
+
36
  def load_model():
37
  tokenizer = DistilBertTokenizerFast()
38
  model = DistilBertForSequenceClassification()