Update README.md
Browse files
README.md
CHANGED
|
@@ -68,7 +68,7 @@ import torch
|
|
| 68 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
| 69 |
|
| 70 |
# Load the model and tokenizer
|
| 71 |
-
model_name = "
|
| 72 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 73 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 74 |
|
|
|
|
| 68 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
| 69 |
|
| 70 |
# Load the model and tokenizer
|
| 71 |
+
model_name = "AHAAM/B2BERT"
|
| 72 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 73 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 74 |
|