Mr-Vicky-01
/

finetuned_classification_model

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Mr-Vicky-01 commited on Sep 3

Commit

0c8fe45

•

1 Parent(s): 1bd649e

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -8,6 +8,7 @@ tags: []
 from transformers import AutoModelForSequenceClassification, DistilBertTokenizer
 import 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)

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