Add download-verified inference example

README.md

@@ -69,3 +69,30 @@ If you use this model, please cite:
   url={{https://huggingface.co/kojima-lab/molcrawl-protein-sequence-bert-medium}}
 }
 ```
+
+
+## Example Output
+
+End-to-end inference test (downloaded the model from this repo on CPU).
+
+```python
+import torch
+from transformers import AutoTokenizer, AutoModelForMaskedLM
+
+REPO_ID = "kojima-lab/molcrawl-protein-sequence-bert-medium"
+tokenizer = AutoTokenizer.from_pretrained(REPO_ID)
+model = AutoModelForMaskedLM.from_pretrained(REPO_ID)
+model.eval()
+
+sequence = "MKTVRQERLKSIVRILERSKEPVSGAQLAEELSVSR<mask>VIVQDIAYLRSLGYNIVATPRGYVLAGG"
+inputs = tokenizer(sequence, return_tensors="pt")
+mask_index = (inputs["input_ids"][0] == tokenizer.mask_token_id).nonzero(as_tuple=True)[0]
+
+with torch.no_grad():
+    outputs = model(**inputs)
+
+predicted_id = outputs.logits[0, mask_index].argmax(dim=-1)
+predicted_aa = tokenizer.convert_ids_to_tokens(predicted_id.tolist())[0]
+print(f"Predicted amino acid at mask: {predicted_aa}")
+# => Predicted amino acid at mask: W
+```