update readme

README.md

@@ -42,7 +42,7 @@ You can use this model for feature extraction (embeddings) or fine-tune it for d
 Use the code below to get started with the model for embedding extraction:
 
 ```python
-from transformers import AutoTokenizer, AutoModel
+from transformers import AutoModel
 from sequence_models.pretrained import load_model_and_alphabet
 import torch
 
@@ -69,7 +69,7 @@ print('Dynamics:', enriched.dyn)
 Use the code below to get started with the model for downstream prediction tasks:
 
 ```python
-from transformers import AutoTokenizer, AutoModel, T5EncoderModel, T5Tokenizer, AutoConfig
+from transformers import AutoModelForSequenceClassification, AutoConfig
 from sequence_models.pretrained import load_model_and_alphabet
 import torch