Update README.md
Browse files
README.md
CHANGED
@@ -48,7 +48,7 @@ Val loss: 0.0013 | accuracy: 1.0000
|
|
48 |
import torch
|
49 |
from transformers import BertTokenizer, BertForSequenceClassification
|
50 |
|
51 |
-
model_path = '
|
52 |
|
53 |
model = BertForSequenceClassification.from_pretrained(model_path, num_labels=3, problem_type="single_label_classification")
|
54 |
tokenizer = BertTokenizer.from_pretrained(model_path)
|
|
|
48 |
import torch
|
49 |
from transformers import BertTokenizer, BertForSequenceClassification
|
50 |
|
51 |
+
model_path = 'alimboff/rubert-kbd-krc-36K'
|
52 |
|
53 |
model = BertForSequenceClassification.from_pretrained(model_path, num_labels=3, problem_type="single_label_classification")
|
54 |
tokenizer = BertTokenizer.from_pretrained(model_path)
|