Update README.md
Browse files
README.md
CHANGED
|
@@ -54,7 +54,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
| 54 |
import torch
|
| 55 |
|
| 56 |
# Load model and tokenizer
|
| 57 |
-
model_checkpoint = 'AdamCodd/
|
| 58 |
model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint)
|
| 59 |
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
|
| 60 |
|
|
|
|
| 54 |
import torch
|
| 55 |
|
| 56 |
# Load model and tokenizer
|
| 57 |
+
model_checkpoint = 'AdamCodd/distilroberta-NLI'
|
| 58 |
model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint)
|
| 59 |
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
|
| 60 |
|