Update README.md

README.md

@@ -50,14 +50,14 @@ Not mentioned parameters are the same as for GPT-2.
 
 ## Usage
 ```python
-from transformers import AutoTokenizer, GPT2LMHeadModel
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
 t = AutoTokenizer.from_pretrained("BUT-FIT/Czech-GPT-2-XL-133k")
-m = GPT2LMHeadModel.from_pretrained("BUT-FIT/Czech-GPT-2-XL-133k").eval()
+m = AutoModelForCausalLM.from_pretrained("BUT-FIT/Czech-GPT-2-XL-133k").eval()
 
 # Try the model inference
-prompt = "Najznámějším českým spisovatelem "
+prompt = "Nejznámějším českým spisovatelem "
 input_ids = t.encode(prompt, return_tensors="pt")
 with torch.no_grad():
     generated_text = m.generate(input_ids=input_ids,