fix model load

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 
-from transformers import AutoTokenizer, DistilBertForSequenceClassification
+from transformers import AutoTokenizer, DistilBertForSequenceClassification, DistilBertConfig
 import torch
 from torch.nn.functional import softmax
 
@@ -28,7 +28,11 @@ id_to_description = load_tags_info()
 
 @st.cache_resource
 def load_model():
-    return DistilBertForSequenceClassification.from_pretrained('./')
+    config = DistilBertConfig.from_json_file('./config.json')
+    model = DistilBertForSequenceClassification(config)
+    state_dict = torch.load('./pytorch_model.bin')
+    model.load_state_dict(state_dict)
+    return model
 
 def load_tokenizer():
     return AutoTokenizer.from_pretrained('distilbert-base-uncased')