Update README.md

README.md

@@ -32,9 +32,9 @@ wget -r -N -c -np --user "INSERT_USER" --ask-password https://physionet.org/file
 
 Then, you can load the model + tokenizer:
 ```
-from transformers import AutoTokenizer, AutoModel
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 tokenizer = AutoTokenizer.from_pretrained(INSERT_PATH_TO_MODEL_FOLDER)
-model = AutoModel.from_pretrained(PATH_TO_MODEL_FOLDER)
+model = AutoModelForSeq2SeqLM.from_pretrained(PATH_TO_MODEL_FOLDER)
 ```
 
 # Questions?