Point README to EmbodiedCity/ANWM.code on GitHub

README.md

@@ -13,7 +13,8 @@ tags:
 Aerial Navigation World Model checkpoint released with the paper
 [Aerial World Model for Long-horizon Visual Generation and Navigation in 3D Space](https://arxiv.org/abs/2512.21887).
 
-Dataset: [EmbodiedCity/ANWM-Dataset](https://huggingface.co/datasets/EmbodiedCity/ANWM-Dataset)
+- **Code**: [https://github.com/EmbodiedCity/ANWM.code](https://github.com/EmbodiedCity/ANWM.code)
+- **Dataset**: [EmbodiedCity/ANWM-Dataset](https://huggingface.co/datasets/EmbodiedCity/ANWM-Dataset)
 
 ## Files
 
@@ -23,9 +24,12 @@ Dataset: [EmbodiedCity/ANWM-Dataset](https://huggingface.co/datasets/EmbodiedCit
 
 ## Quick start
 
-From the ANWM code repo root, download into the path expected by `infer.py`:
+Clone the code, then download this checkpoint into the path expected by `infer.py`:
 
 ```bash
+git clone https://github.com/EmbodiedCity/ANWM.code.git
+cd ANWM.code
+
 huggingface-cli download EmbodiedCity/ANWM 0200000.pth.tar \
   --local-dir logs/anwm_cdit_airvln/checkpoints
 
@@ -49,5 +53,7 @@ logs/anwm_cdit_airvln/checkpoints/0200000.pth.tar
 
 ## Related
 
+- Code: https://github.com/EmbodiedCity/ANWM.code
 - Dataset: https://huggingface.co/datasets/EmbodiedCity/ANWM-Dataset
+- Model: https://huggingface.co/EmbodiedCity/ANWM
 - Paper: https://arxiv.org/abs/2512.21887