diff --git a/README.md b/README.md index 8972045..9289487 100644 --- a/README.md +++ b/README.md @@ -110,14 +110,24 @@ Then import the necessary libraries and instantiate your desired model. ```py from esm.sdk import esmc_client - +from esm.sdk.api import ESMProtein, LogitsConfig + +# Human carbonic anhydrase II (PDB 2CBA) +protein = ESMProtein( + sequence=( + "MSHHWGYGKHNGPEHWHKDFPIAKGERQSPVDIDTHTAKYDPSLKPLSVSYDQATSLRILNNGHAFNVEFDD" + "SQDKAVLKGGPLDGTYRLIQFHFHWGSLDGQGSEHTVDKKKYAAELHLVHWNTKYGDFGKAVQQPDGLAVL" + "GIFLKVGSAKPGLQKVVDVLDSIKTKGKSADFTNFDPRGLLPESLDYWTYPGSLTTPPLLECVTWIVLKEP" + "ISVSSEQVLKFRKLNFNGEGEPEELMVDNWRPAQPLKNRQIKASFK" + ) +) model = esmc_client( model="esmc-600m-2024-12", url="https://biohub.ai", token="" ) protein_tensor = model.encode(protein) logits_output = model.logits( - protein_tensor, LogitsConfig(sequence=True, return_embeddings=True) + protein_tensor, LogitsConfig(sequence=True, return_embeddings=True) ) print(logits_output.logits, logits_output.embeddings) @@ -234,7 +244,8 @@ Import the necessary libraries. ```py from esm.sdk.forge import SequenceStructureForgeInferenceClient -from esm.sdk.api import ESMProtein, ESMProteinError, LogitsConfig, LogitsOutput +from esm.sdk.api import FoldingConfig +from esm.utils.structure.input_builder import ProteinInput, StructurePredictionInput ``` Call the inference client with the selected model of choice and replace with your token name. @@ -242,13 +253,19 @@ Call the inference client with the selected model of choice and replace