diff --git a/README.md b/README.md
index 2db99e0..fd80e1b 100644
--- a/README.md
+++ b/README.md
@@ -33,7 +33,7 @@ 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-base-seq2seq")
 model = AutoModelForSeq2SeqLM.from_pretrained("microsoft/GODEL-v1_1-base-seq2seq")