Update utils.py

utils.py

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