t-bank-ai
/

response-toxicity-classifier-base

Text Classification

Inference Endpoints

Model card Files Files and versions Community

solemn-leader commited on Jul 22, 2022

Commit

9a4ac4b

•

1 Parent(s): c9144de

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -46,7 +46,7 @@ model = AutoModelForSequenceClassification.from_pretrained('tinkoff-ai/response-
 inputs = tokenizer('[CLS]привет[SEP]привет![SEP]как дела?[RESPONSE_TOKEN]норм, у тя как?', max_length=128, add_special_tokens=False, return_tensors='pt')
 with torch.inference_mode():
     logits = model(**inputs).logits
-    probas = torch.sigmoid(logits)[0].cpu().detach().numpy()
 ```

 inputs = tokenizer('[CLS]привет[SEP]привет![SEP]как дела?[RESPONSE_TOKEN]норм, у тя как?', max_length=128, add_special_tokens=False, return_tensors='pt')
 with torch.inference_mode():
     logits = model(**inputs).logits
+    probas = torch.softmax(logits)[0].cpu().detach().numpy()
 ```