Update README.md

README.md

@@ -111,5 +111,14 @@ print(outputs.logits)
   tensor([[-4.3450,  4.6878]], grad_fn=<AddmmBackward0>)
 }
 
+import torch
+
+predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
+print(predictions)
+
+{
+  tensor([[1.1942e-04, 9.9988e-01]], grad_fn=<SoftmaxBackward0>)
+}
+
 
 ```