Update README.md

README.md

@@ -15,11 +15,10 @@ instance_prompt: "Mobile app:"
 # UI-Diffuser-V2
 UI-Diffuser-V2 is fine tuned from "stabilityai/stable-diffusion-2-base" with the [GPSCap dataset](https://github.com/jl-wei/guing) for mobile UI generation.
 
-This iteration, UI-Diffuser-V2, represents the second version of the UI-Diffuser model.
+A demo using diffusion model and large language model for UI generation is available at https://github.com/Jl-wei/ai-gen-ui
 
-The first version, UI-Diffuser-V1, was introduced in our paper titled [Boosting GUI Prototyping with Diffusion Models](https://ieeexplore.ieee.org/abstract/document/10260853)
 
-Using with Diffusers
+## Using with Diffusers
 ```python
 import torch
 from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
@@ -44,4 +43,20 @@ for ax in fig.axes:
     ax.axis("off")
 ```
 
-Please note that the model can only be used for academic purpose.
+## Citation
+If you find our work useful, please cite our paper:
+```bibtex
+@misc{wei2024aiinspired,
+      title={On AI-Inspired UI-Design}, 
+      author={Jialiang Wei and Anne-Lise Courbis and Thomas Lambolais and Gérard Dray and Walid Maalej},
+      year={2024},
+      eprint={2406.13631},
+      archivePrefix={arXiv}
+}
+```
+Please note that the code and model can only be used for academic purpose.
+
+### UI-Diffuser-V1
+This model, UI-Diffuser-V2, represents the second version of the UI-Diffuser model.
+
+The initial version, UI-Diffuser-V1, was introduced in our paper titled [Boosting GUI Prototyping with Diffusion Models](https://ieeexplore.ieee.org/abstract/document/10260853)