Update README.md

README.md

@@ -28,9 +28,8 @@ To evaluate the model, we sample 50 times from the decoder for each input and th
 One can load this model for their personal use of fine-tuning as follows:
 
 ```python
-# Load model directly
-from transformers import AutoModel, AutoTokenizer
-tokenizer = AutoTokenizer.from_pretrained("t5-small")
-tokenizer.add_special_tokens({'sep_token': '[SEP]'})
-model = AutoModel.from_pretrained("DeepPavlov/t5-wikidata5M-with-neighbors")
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+tokenizer = AutoTokenizer.from_pretrained("DeepPavlov/t5-wikidata5M-with-neighbors")
+model = AutoModelForSeq2SeqLM.from_pretrained("DeepPavlov/t5-wikidata5M-with-neighbors")
 ```