Update README.md

README.md

@@ -1,11 +1,20 @@
 ---
 license: apache-2.0
+task_categories:
+- image-to-video
+tags:
+- minecraft
+- world-model
+- multi-agent
 ---
 
 
-
 ![Banner](./solaris-logo.png)
 
+[**Project Page**](https://solaris-wm.github.io/) | [**Paper**](https://huggingface.co/papers/2602.22208) | [**Github**](https://github.com/solaris-wm/solaris)
+
+# Solaris Model
+
 This model repository contains the final model weights of the Solaris Minecraft multiplayer model, along with the intermediate model weights for all training stages and third-party models required for inference and training. All weights are in JAX. Refer to the [Solaris](https://github.com/solaris-wm/solaris) GitHub repository for downloading and running instructions. 
 
 Below is the description of the model folders:
@@ -18,4 +27,15 @@ Below is the description of the model folders:
 - `sp_bidirectional_pretrain_120000.pt`: Single player bidirectional model weights produced by training Stage 1
 - `matrix-game-init.pt`: [Matrix Game 2.0 weights](https://huggingface.co/Skywork/Matrix-Game-2.0/tree/main/base_distilled_model) with the keyboard action MLP dimension increased to 23 (randmoly initialized). Used for single player pretraining (Stage 1) initialization
 
-See [Solaris](https://github.com/solaris-wm/solaris?tab=readme-ov-file#training-stages) repo for a detailed description of the training stages.
+See [Solaris](https://github.com/solaris-wm/solaris?tab=readme-ov-file#training-stages) repo for a detailed description of the training stages.
+
+# Citation
+
+```bibtex
+@article{savva2025solaris,
+  title={Solaris: Building a Multiplayer Video World Model in Minecraft},
+  author={Georgy Savva and Oscar Michel and Daohan Lu and Suppakit Waiwitlikhit and Timothy Meehan and Dhairya Mishra and Srivats Poddar and Jack Lu and Saining Xie},
+  journal={arXiv preprint arXiv:2602.22208},
+  year={2025}
+}
+```