longchen0421
commited on
Commit
•
bf720af
1
Parent(s):
4792c7f
Update README.md
Browse files
README.md
CHANGED
@@ -37,7 +37,7 @@ Here is a simple code for inference:
|
|
37 |
```python
|
38 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
39 |
|
40 |
-
model_name = 'plant-dnagemma-
|
41 |
# load model and tokenizer
|
42 |
model = AutoModelForSequenceClassification.from_pretrained(f'zhangtaolab/{model_name}', trust_remote_code=True)
|
43 |
tokenizer = AutoTokenizer.from_pretrained(f'zhangtaolab/{model_name}', trust_remote_code=True)
|
|
|
37 |
```python
|
38 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
39 |
|
40 |
+
model_name = 'plant-dnagemma-singlebase-conservation'
|
41 |
# load model and tokenizer
|
42 |
model = AutoModelForSequenceClassification.from_pretrained(f'zhangtaolab/{model_name}', trust_remote_code=True)
|
43 |
tokenizer = AutoTokenizer.from_pretrained(f'zhangtaolab/{model_name}', trust_remote_code=True)
|