Change loaded mode to correct on that this model card refers to
#26
by
Liqs
- opened
README.md
CHANGED
@@ -141,8 +141,8 @@ Example of single-label classification:
|
|
141 |
import torch
|
142 |
from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
|
143 |
|
144 |
-
tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
|
145 |
-
model = DistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased")
|
146 |
|
147 |
inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
|
148 |
with torch.no_grad():
|
|
|
141 |
import torch
|
142 |
from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
|
143 |
|
144 |
+
tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
|
145 |
+
model = DistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
|
146 |
|
147 |
inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
|
148 |
with torch.no_grad():
|