update how to use

README.md

@@ -18,16 +18,16 @@ For code and evaluation check out [GitHub](https://github.com/readerbench/RoGPT2
 
 ```python
 # TensorFlow
-from transformers import AutoTokenizer, TFAutoModelWithLMHead
+from transformers import AutoTokenizer, TFAutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained('readerbench/RoGPT2-large')
-model = TFAutoModelWithLMHead.from_pretrained('readerbench/RoGPT2-large')
+model = TFAutoModelForCausalLM.from_pretrained('readerbench/RoGPT2-large')
 inputs = tokenizer.encode("Este o zi de vara", return_tensors='tf')
 text = model.generate(inputs, max_length=1024,  no_repeat_ngram_size=2)
 print(tokenizer.decode(text[0]))
 
 # PyTorch
-from transformers import AutoTokenizer, AutoModelWithLMHead
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained('readerbench/RoGPT2-large')
 model = AutoModelForCausalLM.from_pretrained('readerbench/RoGPT2-large')