RaphaelMourad
/

Mistral-Peptide-v1-15M

Model card Files Files and versions Community

RaphaelMourad commited on 5 days ago

Commit

8b7cc1d

•

1 Parent(s): e7c188c

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -33,8 +33,8 @@ model = AutoModel.from_pretrained("RaphaelMourad/Mistral-Peptide-v1-15M", trust_
 ## Calculate the embedding of a protein sequence
 ```
-insulin = "MALWMRLLPLLALLALWG"
-inputs = tokenizer(insulin, return_tensors = 'pt')["input_ids"]
 hidden_states = model(inputs)[0] # [1, sequence_length, 256]
 # embedding with max pooling

 ## Calculate the embedding of a protein sequence
 ```
+peptide = "MALWMRLLPLLALLALWG"
+inputs = tokenizer(peptide, return_tensors = 'pt')["input_ids"]
 hidden_states = model(inputs)[0] # [1, sequence_length, 256]
 # embedding with max pooling