Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import torch
 from transformers import DistilBertForSequenceClassification, DistilBertTokenizerFast
-from utils import get_text, get_labels
+from utils import get_text, get_labels, load_model
 
 count_labels = 8
 
@@ -15,8 +15,9 @@ abstract = st.text_area("Введите аннотацию к статье, abst
 
 tokenizer = DistilBertTokenizerFast()
 model = DistilBertForSequenceClassification()
-@st.cache
-model.load_state_dict(torch.load('weight_model'))
+
+load_model(model, 'weight_model')
+#model.load_state_dict(torch.load('weight_model'))
 
 text = get_text(title, abstract)
 if text: