Update README.md
Browse files
README.md
CHANGED
@@ -21,7 +21,7 @@ from transformers import BertForSequenceClassification, BertTokenizer
|
|
21 |
|
22 |
# Step 1: Load the model and tokenizer from Hugging Face
|
23 |
model = BertForSequenceClassification.from_pretrained("rahulm-selector/log-classifier-BERT-v1")
|
24 |
-
tokenizer = BertTokenizer.from_pretrained("
|
25 |
|
26 |
import torch
|
27 |
|
|
|
21 |
|
22 |
# Step 1: Load the model and tokenizer from Hugging Face
|
23 |
model = BertForSequenceClassification.from_pretrained("rahulm-selector/log-classifier-BERT-v1")
|
24 |
+
tokenizer = BertTokenizer.from_pretrained("SelectorAI/log-classifier-BERT-v1")
|
25 |
|
26 |
import torch
|
27 |
|