Commit
•
af50bb1
1
Parent(s):
ec7cc43
Update README.md
Browse files
README.md
CHANGED
@@ -16,7 +16,7 @@ This is a small sequence classifier trained on synthetic data generate by GPT-4
|
|
16 |
which classifies protein sequences into three categories `enzymes`, `transport_proteins`, and `structural_proteins`.
|
17 |
To use the model, try running:
|
18 |
|
19 |
-
```
|
20 |
# Load the trained model and tokenizer
|
21 |
model = EsmForSequenceClassification.from_pretrained("./esm2_t6_8M_UR50D_sequence_classifier_v1")
|
22 |
tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t6_8M_UR50D")
|
|
|
16 |
which classifies protein sequences into three categories `enzymes`, `transport_proteins`, and `structural_proteins`.
|
17 |
To use the model, try running:
|
18 |
|
19 |
+
``` python3
|
20 |
# Load the trained model and tokenizer
|
21 |
model = EsmForSequenceClassification.from_pretrained("./esm2_t6_8M_UR50D_sequence_classifier_v1")
|
22 |
tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t6_8M_UR50D")
|