arkmartov commited on
Commit
9c81189
·
1 Parent(s): 8175f03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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
  # выводим результаты модели в текстовое поле, на потеху пользователю