use AutoModelForCausalLM in model card example

README.md

@@ -33,9 +33,9 @@ Train loss: ~2.18 · Dev loss: ~2.20
 ```python
 import torch
 import torch.nn.functional as F
-from transformers import AutoModel, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model = AutoModel.from_pretrained("iand666/makemore-mlp", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("iand666/makemore-mlp", trust_remote_code=True)
 tok   = AutoTokenizer.from_pretrained("iand666/makemore-mlp", trust_remote_code=True)
 model.eval()