Update README.md

README.md

@@ -24,6 +24,7 @@ Please download the checkpoints and organize them as the following file structur
         |-- ckpt_carry.pth 
     |-- tokenhsi
         |-- ckpt_stage1.pth
+        |-- ckpt_stage2_longterm.pth
         |-- ...
         |-- (more models are coming soon!)
 ```
@@ -49,7 +50,7 @@ Please download the checkpoints and organize them as the following file structur
      * Follow: tokenhsi/ckpt_stage2_terrainShape_traj.pth
      * Carry: tokenhsi/ckpt_stage2_terrainShape_carry.pth
   * Long-horizon Task Completion 
-     * tokenhsi/ckpt_stage2_long_task.pth
+     * tokenhsi/ckpt_stage2_longterm.pth
 
 GitHub repo: https://github.com/liangpan99/TokenHSI  
 arXiv paper: https://huggingface.co/papers/2503.19901