Update README.md

README.md

@@ -20,9 +20,9 @@ BashGPT-Neo is a [Neural Program Synthesis](https://www.microsoft.com/en-us/rese
 
 ## Usage
 ```py
-from transformers import AutoTokenizer, AutoModelWithLMHead
+from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("reshinthadith/BashGPTNeo")
-model = AutoModelWithLMHead.from_pretrained("reshinthadith/BashGPTNeo")
+model = AutoModelForCausalLM.from_pretrained("reshinthadith/BashGPTNeo")
 ```
 
 ## Core Contributors 👥