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",
|