Mr-Vicky-01
/

finetuned_classification_model

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Mr-Vicky-01 commited on Aug 28

Commit

9920e89

•

1 Parent(s): 36056e5

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -10,8 +10,8 @@ import time
 import torch
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-model = AutoModelForSequenceClassification.from_pretrained("finetuned_classification_model").to(device)
-tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
 start = time.time()

 import torch
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+model = AutoModelForSequenceClassification.from_pretrained("Mr-Vicky-01/finetuned_classification_model").to(device)
+tokenizer = DistilBertTokenizer.from_pretrained("Mr-Vicky-01/finetuned_classification_model")
 start = time.time()