Update README.md

README.md

@@ -8,5 +8,28 @@ datasets:
 library_name: timm
 ---
 
-- Code: https://github.com/sinahmr/LocAtViT
-- Paper: https://arxiv.org/abs/2603.04892
+# LocAtViT: Locality-Attending Vision Transformer
+
+[![arXiv](https://img.shields.io/badge/arXiv-2603.04892-b31b1b.svg)](https://arxiv.org/abs/2603.04892)
+[![hfpaper](https://img.shields.io/badge/GitHub-LocAtViT-black)](https://github.com/sinahmr/LocAtViT)
+
+> Pretrain vision transformers so that their patch representations transfer better to dense prediction (e.g., segmentation), without changing the pretraining objective.
+
+
+## Usage
+
+```python
+import timm
+model = timm.create_model("hf_hub:sinahmr/locatvit_base", pretrained=True)
+```
+
+## Citation
+
+```bibtex
+@inproceedings{hajimiri2026locatvit,
+  author    = {Hajimiri, Sina and Beizaee, Farzad and Shakeri, Fereshteh and Desrosiers, Christian and Ben Ayed, Ismail and Dolz, Jose},
+  title     = {Locality-Attending Vision Transformer},
+  booktitle = {International Conference on Learning Representations},
+  year      = {2026}
+}
+```