SaeedLab
/

ProtEnrich-Ankh3

Feature Extraction

Model card Files Files and versions

gabrielbianchin commited on Mar 3

Commit

e184c85

·

1 Parent(s): 530a6a6

update readme

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -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
 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
 import torch
 tokenizer = T5Tokenizer.from_pretrained("ElnaggarLab/ankh3-xl")

 Use the code below to get started with the model for embedding extraction:
 ```python
+from transformers import AutoModel, T5EncoderModel, T5Tokenizer
 import torch
 tokenizer = T5Tokenizer.from_pretrained("ElnaggarLab/ankh3-xl")
 Use the code below to get started with the model for downstream prediction tasks:
 ```python
+from transformers import AutoModelForSequenceClassification, T5EncoderModel, T5Tokenizer, AutoConfig
 import torch
 tokenizer = T5Tokenizer.from_pretrained("ElnaggarLab/ankh3-xl")