Add inference example output to model card

README.md

@@ -69,3 +69,33 @@ If you use this model, please cite:
   url={{https://huggingface.co/kojima-lab/molcrawl-genome-sequence-bert-small}}
 }
 ```
+
+
+## Example Output
+
+Inference test performed on the uploaded checkpoint (CPU):
+
+```python
+from transformers import AutoModelForMaskedLM, AutoTokenizer
+import torch
+
+model = AutoModelForMaskedLM.from_pretrained("kojima-lab/molcrawl-genome-sequence-bert-small")
+tokenizer = AutoTokenizer.from_pretrained("kojima-lab/molcrawl-genome-sequence-bert-small")
+
+# DNA masked token prediction (SentencePiece BPE subword vocabulary)
+# Mask token: '[MASK]'
+prompt = "ATCGATCGATCG[MASK]GCTTATCAAGCT"
+inputs = tokenizer(prompt, return_tensors="pt")
+mask_token_id = tokenizer.mask_token_id
+mask_index = (inputs["input_ids"] == mask_token_id).nonzero(as_tuple=True)[1]
+
+with torch.no_grad():
+    outputs = model(**inputs)
+logits = outputs.logits
+
+predicted_token_id = logits[0, mask_index].argmax(dim=-1)
+predicted_token = tokenizer.decode(predicted_token_id)
+result = prompt.replace(tokenizer.mask_token, predicted_token)
+print(f"Predicted: {result}")
+# => Predicted: ATCGATCGATCGCGCTTATCAAGCT
+```