Update README.md
Browse files
README.md
CHANGED
|
@@ -52,10 +52,10 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
| 52 |
import torch
|
| 53 |
|
| 54 |
# Load model and tokenizer
|
| 55 |
-
|
| 56 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
| 57 |
model = AutoModelForSequenceClassification.from_pretrained(
|
| 58 |
-
|
| 59 |
num_labels=1,
|
| 60 |
problem_type="regression"
|
| 61 |
)
|
|
|
|
| 52 |
import torch
|
| 53 |
|
| 54 |
# Load model and tokenizer
|
| 55 |
+
|
| 56 |
+
tokenizer = AutoTokenizer.from_pretrained("DarianNLP/bert_sequel_beagles")
|
| 57 |
model = AutoModelForSequenceClassification.from_pretrained(
|
| 58 |
+
"DarianNLP/bert_sequel_beagles",
|
| 59 |
num_labels=1,
|
| 60 |
problem_type="regression"
|
| 61 |
)
|