Change loaded mode to correct on that this model card refers to

#26
by Liqs - opened
Files changed (1) hide show
  1. README.md +2 -2
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():