Update utils.py

utils.py

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