Update utils.py

utils.py

@@ -31,7 +31,7 @@ def get_labels(text, model, tokenizer, count_labels=8):
           return result_labels
       result_labels.append(pair[1])
  
- @st.cache()
+ @st.cache(allow_output_mutation=True)
  def load_model(model, filename):
    model.load_state_dict(torch.load(filename)) 
    return model