Update utils.py

utils.py

@@ -34,8 +34,8 @@ def get_labels(text, model, tokenizer, count_labels=8):
  
 @st.cache(allow_output_mutation=True)
 def load_model():
-   tokenizer = DistilBertTokenizerFast
-   model = DistilBertForSequenceClassification
+   tokenizer = DistilBertTokenizerFast.from_pretrained("distilbert-base-cased")
+   model = DistilBertForSequenceClassification.from_pretrained("distilbert-base-cased", num_labels=8)
    model.load_state_dict(torch.load('weight_model')) 
    return model, tokenizer