Update README.md

README.md

@@ -31,6 +31,7 @@ logits = outputs.logits
 # model predicts one of the 1000 ImageNet classes
 predicted_class_idx = logits.argmax(-1).item()
 print("Predicted class:", model.config.id2label[predicted_class_idx])
+# Predicted class: Egyptian cat
 ```
 
 # Citation