Alanhau
/

test

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Alanhau commited on Feb 28, 2023

Commit

12afa9c

•

1 Parent(s): 8b9324f

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -140,8 +140,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("Alanhau/test")
+model = DistilBertForSequenceClassification.from_pretrained("Alanhau/test")
 inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
 with torch.no_grad():