Update README.md

README.md

@@ -76,7 +76,7 @@ tensor([[    0.004,     0.002,     0.701,     0.014,     0.030,     0.011,
              0.000,     0.008,     0.004,     0.001,     0.002,     0.009,
              0.004,     0.002,     0.005,     0.008,     0.001,     0.005]],
        grad_fn=<SoftmaxBackward0>)
->>> top_class = torch.argmax(logits.logits, dim = -1)
+>>> top_class = torch.argmax(logits.logits, dim=1)
 >>> top_class
 tensor([2])
 >>> model.config.id2label[top_class.numpy()[0]]