AmelieSchreiber
/

PepMLM_v0

Model card Files Files and versions

AmelieSchreiber commited on Nov 22, 2023

Commit

2d760c3

·

1 Parent(s): 27d2c5d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -83,7 +83,7 @@ def generate_peptide(input_seqs, peptide_length=15, top_k=3, num_binders=4):
                 results.append([seq, binder, ppl])
         return pd.DataFrame(results, columns=['Input Sequence', 'Binder', 'Pseudo Perplexity'])
-model = EsmForMaskedLM.from_pretrained("output/checkpoint-255")
 tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t33_650M_UR50D")
 protein_seq = "MAPLRKTYVLKLYVAGNTPNSVRALKTLNNILEKEFKGVYALKVIDVLKNPQLAEEDKILATPTLAKVLPPPVRRIIGDLSNREKVLIGLDLLYEEIGDQAEDDLGLE"

                 results.append([seq, binder, ppl])
         return pd.DataFrame(results, columns=['Input Sequence', 'Binder', 'Pseudo Perplexity'])
+model = EsmForMaskedLM.from_pretrained("AmelieSchreiber/PepMLM_v0")
 tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t33_650M_UR50D")
 protein_seq = "MAPLRKTYVLKLYVAGNTPNSVRALKTLNNILEKEFKGVYALKVIDVLKNPQLAEEDKILATPTLAKVLPPPVRRIIGDLSNREKVLIGLDLLYEEIGDQAEDDLGLE"