Update README.md
Browse files
README.md
CHANGED
|
@@ -47,8 +47,8 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
|
| 47 |
import torch
|
| 48 |
|
| 49 |
# Load model and tokenizer
|
| 50 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 51 |
-
model = AutoModelForSequenceClassification.from_pretrained("
|
| 52 |
|
| 53 |
# Example text
|
| 54 |
text = "He put the phone back in the cradle and turned toward the kitchen."
|
|
|
|
| 47 |
import torch
|
| 48 |
|
| 49 |
# Load model and tokenizer
|
| 50 |
+
tokenizer = AutoTokenizer.from_pretrained("ChangeIsKey/text-dating")
|
| 51 |
+
model = AutoModelForSequenceClassification.from_pretrained("ChangeIsKey/text-dating")
|
| 52 |
|
| 53 |
# Example text
|
| 54 |
text = "He put the phone back in the cradle and turned toward the kitchen."
|