Update app.py

app.py

@@ -4,9 +4,11 @@ from transformers import BertTokenizer
 import gradio as gr
 from model import BertClassifier
 
-# Загрузка модели
-model = BertClassifier(dropout_rate=0.3, num_classes=4)
-model.load_state_dict(torch.load("bert_mc_dropout.pt", map_location="cpu"))
+model = BertClassifier()
+state_dict = torch.load("bert_mc_dropout.pt", map_location="cpu")
+
+# подгружаем корректно
+model.load_state_dict(state_dict, strict=False)
 model.eval()
 
 # Токенизатор