Hugging Face's logo

Spaces:
Dzhamb
/
ArticleClassification
Runtime error

App Files Community
Dzhamb commited on
Commit
7a6d4f6
·
1 Parent(s): c2c3782

Update utils.py

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