Update README.md

README.md

@@ -21,6 +21,13 @@ Interpretable slot-based attention achieving competitive language modeling perfo
 
 from asa import load_asm_checkpoint, generate
 from transformers import AutoTokenizer
+from huggingface_hub import hf_hub_download
+
+# Download checkpoint from Hugging Face
+ckpt_path = hf_hub_download(
+    repo_id="DigitalDaimyo/AddressedStateAttention",
+    filename="checkpoint.pt"   # adjust if your filename differs
+)
 
 # Load checkpoint
 model, cfg, ckpt = load_asm_checkpoint(