Update README.md

README.md

@@ -66,7 +66,7 @@ SequenceClassifierOutput(loss=None, logits=tensor([[-0.4635, -1.0112,  4.7935,
          -0.1097, -1.8827,  0.6609, -0.2022,  0.0971, -0.6205,  0.4492,  0.0926,
          -2.4848,  0.2630, -0.4584, -2.4327, -1.1654,  0.3897, -0.3374, -1.2418,
          -0.1045,  0.2827, -1.5667, -0.0963]], grad_fn=<AddmmBackward0>), hidden_states=None, attentions=None)
->>> classes = torch.softmax(logits.logits, dim = -1)
+>>> classes = torch.softmax(logits.logits, dim=1)
 >>> torch.set_printoptions(precision=3, sci_mode=False)
 >>> classes
 tensor([[    0.004,     0.002,     0.701,     0.014,     0.030,     0.011,