Update app.py

app.py

@@ -13,8 +13,8 @@ title = st.text_area("Введите название статьи")
 
 abstract = st.text_area("Введите аннотацию к статье, abstract статьи")
 
-tokenizer = DistilBertTokenizerFast.from_pretrained('distilbert-base-cased')
-model = DistilBertForSequenceClassification.from_pretrained("distilbert-base-cased", num_labels=count_labels)
+tokenizer = DistilBertTokenizerFast()
+model = DistilBertForSequenceClassification()
 model.load_state_dict(torch.load('weight_model'))
 
 text = get_text(title, abstract)