Improve model card and add paper links

README.md

@@ -1,5 +1,6 @@
 ---
 license: apache-2.0
+pipeline_tag: text-to-image
 tags:
 - text-to-image
 - image-generation
@@ -9,7 +10,11 @@ tags:
 
 # GridAR-3B-T2I
 
-Autoregressive Text-to-Image Model with 2D Grid Tokens (a Controlled Baseline for FlexTok)
+Autoregressive Text-to-Image Model with 2D Grid Tokens (a Controlled Baseline for FlexTok).
+
+This model was presented in the paper **[(1D) Ordered Tokens Enable Efficient Test-Time Search](https://huggingface.co/papers/2604.15453)**.
+
+[**Project Page**](https://soto.epfl.ch/) | [**GitHub Repository**](https://github.com/EPFL-VILAB/search-over-tokens)
 
 ## Model Details
 
@@ -19,11 +24,11 @@ Autoregressive Text-to-Image Model with 2D Grid Tokens (a Controlled Baseline fo
 - **Number of Blocks**: 36
 - **Number of Heads**: 36
 - **Image Resolution**: 256x256
-- **Tokenizer**: ZhitongGao/GridAR_256
+- **Tokenizer**: [ZhitongGao/GridAR_256](https://huggingface.co/ZhitongGao/GridAR_256)
 - **Text Encoder**: google/flan-t5-xl
 
 ## Installation
-For install instructions, please see https://github.com/EPFL-VILAB/search-over-tokens/.
+For install instructions, please see the [official repository](https://github.com/EPFL-VILAB/search-over-tokens/).
 
 ## Usage
 
@@ -52,15 +57,15 @@ If you find this repository helpful, please consider citing our work:
 ```bibtex
 @article{gao2026ordered,
   title={(1D) Ordered Tokens Enable Efficient Test-Time Search},
-  author={Zhitong Gao and Parham Rezaei and Ali Cy and Mingqiao Ye and Nata{\v{s}}a Jovanovi{\'{c}} and Jesse Allardice and Afshin Dehghan and Amir Zamir and Roman Bachmann and O{\u{g}}uzhan Fatih Kar},
+  author={Zhitong Gao and Parham Rezaei and Ali Cy and Mingqiao Ye and Nata{\v{s}}a Jovanovi{\\'{c}} and Jesse Allardice and Afshin Dehghan and Amir Zamir and Roman Bachmann and O{\\u{g}}uzhan Fatih Kar},
   journal={arxiv 2026},
   year={2026}
 }
 
 @article{flextok,
   title={{FlexTok}: Resampling Images into 1D Token Sequences of Flexible Length},
-  author={Roman Bachmann and Jesse Allardice and David Mizrahi and Enrico Fini and O{\u{g}}uzhan Fatih Kar and Elmira Amirloo and Alaaeldin El-Nouby and Amir Zamir and Afshin Dehghan},
+  author={Roman Bachmann and Jesse Allardice and David Mizrahi and Enrico Fini and O{\\u{g}}uzhan Fatih Kar and Elmira Amirloo and Alaaeldin El-Nouby and Amir Zamir and Afshin Dehghan},
   journal={arXiv 2025},
   year={2025}
 }
-```
+```