Update README.md (#2)

- Update README.md (87df53c645a1612c3b79fd545deb57c6ac29c738)

Co-authored-by: Sameer Ahmed <SameerAhmed-7@users.noreply.huggingface.co>

Files changed (1) hide show

README.md CHANGED Viewed

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

 with torch.no_grad():
   prediction = model(**x)
+label = torch.argmax(prediction[0]).item()
 print(f"Label: {label}")
 ```