Update README.md

README.md

@@ -18,11 +18,11 @@ Here is the script that I used to enable the 4 line chat:
 
 ```python
 import torch
-from transformers import AutoTokenizer, AutoModelWithLMHead
-  
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
-# Download model from HuggingFace
-model = AutoModelWithLMHead.from_pretrained('vuminhtue/DialoGPT-large-HP')
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("vuminhtue/DialoGPT-large-HarryPotter3")
+
+model = AutoModelForCausalLM.from_pretrained("vuminhtue/DialoGPT-large-HarryPotter3")
 
 
 # Let's chat for 4 lines