Update README.md (#2)

- Update README.md (87df53c645a1612c3b79fd545deb57c6ac29c738)


Co-authored-by: Sameer Ahmed <[email protected]>

README.md

@@ -41,7 +41,7 @@ model.eval()
 with torch.no_grad():
   prediction = model(**x)
 
-label = torch.argmax(outputs[0]).item()
+label = torch.argmax(prediction[0]).item()
 
 print(f"Label: {label}")
 ```