Update utils.py

utils.py

@@ -1,4 +1,5 @@
 import torch
+from transformers import DistilBertForSequenceClassification, DistilBertTokenizerFast
 import streamlit as st
 
 
@@ -33,6 +34,8 @@ def get_labels(text, model, tokenizer, count_labels=8):
  
  @st.cache(allow_output_mutation=True)
  def load_model():
+   tokenizer = DistilBertTokenizerFast()
+   model = DistilBertForSequenceClassification()
    model.load_state_dict(torch.load('weight_model')) 
    return model