diff --git a/README.md b/README.md
index b90e22f..bddf574 100644
--- a/README.md
+++ b/README.md
@@ -32,7 +32,7 @@ ArXiv paper: [https://arxiv.org/abs/2206.11309](https://arxiv.org/abs/2206.11309
 Now we are ready to try out how the model works as a chatting partner!
 
 ```python
-from transformers import AutoTokenizer,AutoModel
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 tokenizer = AutoTokenizer.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
 model = AutoModelForSeq2SeqLM.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
 def generate(instruction, knowledge, dialog):