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