Update README.md

README.md

@@ -56,3 +56,12 @@ predicted_class_idx = logits.argmax(-1).item()
 predicted_class_label = test_dataset.classes[predicted_class_idx]
 print(f"Predicted class: '{predicted_class_label}', confidence: {logits[0, predicted_class_idx]:.2f}")
 ```
+
+The output of above code snippets should be like:
+
+```bash
+Ground truth class: 'cat'
+tensor([[-1.1497, -0.1080, -0.7349,  9.2517, -1.3094,  0.5403, -0.9521, -1.0223,
+         -1.4102, -1.5389]], grad_fn=<AddmmBackward0>)
+Predicted class: 'cat', confidence: 9.25
+```