Spaces:

Dzhamb
/

ArticleClassification

Runtime error

Dzhamb commited on Mar 26, 2022

Commit

ab282f2

1 Parent(s): 7a6d4f6

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -33,5 +33,6 @@ def get_labels(text, model, tokenizer, count_labels=8):
  @st.cache()
  def load_model(model, filename):
-   return model.load_state_dict(torch.load(filename))

  @st.cache()
  def load_model(model, filename):
+   model.load_state_dict(torch.load(filename))
+   return model