Update README.md

README.md

@@ -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.softmax(logits)[0].cpu().detach().numpy()
+    probas = torch.softmax(logits, dim=-1)[0].cpu().detach().numpy()
 ```