SaeedLab
/

ProtEnrich-ProGen2

Feature Extraction

Model card Files Files and versions

gabrielbianchin commited on 7 days ago

Commit

eef3da7

·

1 Parent(s): 00c0ecc

upload

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 tokenizer = AutoTokenizer.from_pretrained('hugohrban/progen2-base', trust_remote_code=True)
 encoder = AutoModelForCausalLM.from_pretrained("hugohrban/progen2-base",  trust_remote_code=True)
-protenrich = AutoModel.from_pretrained("SaeedLab/ProtEnrich-ProtBERT", trust_remote_code=True)
 seqs = ["MKTFFVLLL"]
 inputs = tokenizer(seqs, return_tensors="pt", padding=True)

 tokenizer = AutoTokenizer.from_pretrained('hugohrban/progen2-base', trust_remote_code=True)
 encoder = AutoModelForCausalLM.from_pretrained("hugohrban/progen2-base",  trust_remote_code=True)
+protenrich = AutoModel.from_pretrained("SaeedLab/ProtEnrich-ProGen2", trust_remote_code=True)
 seqs = ["MKTFFVLLL"]
 inputs = tokenizer(seqs, return_tensors="pt", padding=True)