SaeedLab
/

ProtEnrich-ESM1b

Feature Extraction

Model card Files Files and versions

gabrielbianchin commited on Mar 3

Commit

bfcd684

·

1 Parent(s): 77ca842

upload

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 tokenizer = AutoTokenizer.from_pretrained('facebook/esm1b_t33_650M_UR50S')
 encoder = AutoModel.from_pretrained("facebook/esm1b_t33_650M_UR50S")
-protenrich = AutoModel.from_pretrained("SaeedLab/ProtEnrich-ProtBERT", trust_remote_code=True)
 seqs = ["MKTFFVLLL"]
 seqs = [" ".join(i) for i in seqs]

 tokenizer = AutoTokenizer.from_pretrained('facebook/esm1b_t33_650M_UR50S')
 encoder = AutoModel.from_pretrained("facebook/esm1b_t33_650M_UR50S")
+protenrich = AutoModel.from_pretrained("SaeedLab/ProtEnrich-ESM1b", trust_remote_code=True)
 seqs = ["MKTFFVLLL"]
 seqs = [" ".join(i) for i in seqs]