Update app.py
Browse files
app.py
CHANGED
@@ -18,5 +18,7 @@ from transformers import BertTokenizer, BertForSequenceClassification
|
|
18 |
model_name = "google/bert_uncased_L-4_H-256_A-4"
|
19 |
tokenizer = BertTokenizer.from_pretrained(model_name)
|
20 |
model = BertForSequenceClassification.from_pretrained(model_name, num_labels=8)
|
|
|
|
|
21 |
st.markdown(f"{model}")
|
22 |
# выводим результаты модели в текстовое поле, на потеху пользователю
|
|
|
18 |
model_name = "google/bert_uncased_L-4_H-256_A-4"
|
19 |
tokenizer = BertTokenizer.from_pretrained(model_name)
|
20 |
model = BertForSequenceClassification.from_pretrained(model_name, num_labels=8)
|
21 |
+
bn_state_dict = torch.load('model_w.pt')
|
22 |
+
model.load_state_dict(bn_state_dict)
|
23 |
st.markdown(f"{model}")
|
24 |
# выводим результаты модели в текстовое поле, на потеху пользователю
|