Update README.md
Browse files
README.md
CHANGED
@@ -143,7 +143,7 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
|
143 |
import torch
|
144 |
|
145 |
# Load the fine-tuned model and tokenizer
|
146 |
-
model_name = "ymoslem/ModernBERT-large-qe-v1"
|
147 |
model = AutoModelForSequenceClassification.from_pretrained(
|
148 |
model_name,
|
149 |
device_map="auto",
|
|
|
143 |
import torch
|
144 |
|
145 |
# Load the fine-tuned model and tokenizer
|
146 |
+
model_name = "ymoslem/ModernBERT-large-qe-maxlen512-v1"
|
147 |
model = AutoModelForSequenceClassification.from_pretrained(
|
148 |
model_name,
|
149 |
device_map="auto",
|