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, T5EncoderModel, T5Tokenizer
+from transformers import AutoModel, T5EncoderModel, T5Tokenizer
 import torch
 
 tokenizer = T5Tokenizer.from_pretrained("ElnaggarLab/ankh3-xl")
@@ -70,7 +70,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, T5EncoderModel, T5Tokenizer, AutoConfig
 import torch
 
 tokenizer = T5Tokenizer.from_pretrained("ElnaggarLab/ankh3-xl")