Mr-Vicky-01
/

helping_agent_classification

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Mr-Vicky-01 commited on 2 days ago

Commit

c141be8

•

1 Parent(s): 04437e0

Update README.md

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -9,7 +9,6 @@ tags: []
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 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/helping_agent_classification").to(device)

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