Update README.md

README.md

@@ -47,11 +47,12 @@ For a list of individual model parameters, refer to the `config.json` file in th
 
 ## Using the models
 
+
 ```python
-from transformers import T5Tokenizer, T5ForConditionalGeneration
+from transformers import AutoTokenzier, AutoModelForSeq2SeqLM
 
-tokenizer = T5Tokenizer.from_pretrained("gsarti/it5-small")
-model = T5ForConditionalGeneration.from_pretrained("gsarti/it5-small")
+tokenizer = AutoTokenizer.from_pretrained("gsarti/it5-small")
+model = AutoModelForSeq2SeqLM.from_pretrained("gsarti/it5-small")
 ```
 
 *Note: You will need to fine-tune the model on your downstream seq2seq task to use it. See an example [here](https://huggingface.co/it5/it5-base-question-answering).*