distilbert
/

distilbert-base-uncased-finetuned-sst-2-english

Text Classification

Transformers

PyTorch

Model card Files Files and versions Community

lysandre HF staff

Liqs commited on Dec 19, 2023

Commit

714eb0f

•

1 Parent(s): 4643665

Change loaded mode to correct on that this model card refers to (#26)

Browse files

- Change loaded mode to correct on that this model card refers to (6a50985739210b7567055a1414bf2449cc241c1b)

Co-authored-by: Tobias Lindenbauer <Liqs@users.noreply.huggingface.co>

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -141,8 +141,8 @@ Example of single-label classification:
 import torch
 from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
-tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
-model = DistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased")
 inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
 with torch.no_grad():

 import torch
 from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
+tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
+model = DistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
 inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
 with torch.no_grad():