Update app.py

app.py

@@ -65,12 +65,16 @@ STYLE_NAMES = list(styles.keys())
 MAX_SEED = np.iinfo(np.int32).max
 
 title = r"""
-<h1 align="center">OMG: Occlusion-friendly Personalized Multi-concept Generation In Diffusion Models (OMG + InstantID)</h1>
+<h1 align="center"> OMG + InstantID </h1>
 """
 
 description = r"""
-<b>Official 🤗 Gradio demo</b> for <a href='https://github.com/kongzhecn/OMG/' target='_blank'><b>OMG: Occlusion-friendly Personalized Multi-concept Generation In Diffusion Models</b></a>.<be>.<br>
-<a href='https://kongzhecn.github.io/omg-project/' target='_blank'><b>[Project]</b></a>.<a href='https://github.com/kongzhecn/OMG/' target='_blank'><b>[Code]</b></a>.<a href='https://arxiv.org/abs/2403.10983/' target='_blank'><b>[Arxiv]</b></a>.<br>
+<b>Official 🤗 Gradio demo</b> for <a href='https://github.com/kongzhecn/OMG/' target='_blank'><b>OMG: Occlusion-friendly Personalized Multi-concept Generation In Diffusion Models</b></a>.<be><br>
+<br>
+<a href='https://kongzhecn.github.io/omg-project/' target='_blank'><b>[Project]</b></a><a href='https://github.com/kongzhecn/OMG/' target='_blank'><b>[Code]</b></a><a href='https://arxiv.org/abs/2403.10983/' target='_blank'><b>[Arxiv]</b></a><br>
+<br>
+❗️<b>Related demos<b>:<a href='https://huggingface.co/spaces/Fucius/OMG/' target='_blank'><b> OMG + LoRAs </b></a>❗️<br>
+<br>
 How to use:<br>
 1. Select two characters.
 2. Enter a text prompt as done in normal text-to-image models.
@@ -84,11 +88,11 @@ article = r"""
 <br>
 If our work is helpful for your research or applications, please cite us via:
 ```bibtex
-@article{,
-title={OMG: Occlusion-friendly Personalized Multi-concept Generation In Diffusion Models},
-author={},
-journal={},
-year={}
+@article{kong2024omg,
+  title={OMG: Occlusion-friendly Personalized Multi-concept Generation in Diffusion Models},
+  author={Kong, Zhe and Zhang, Yong and Yang, Tianyu and Wang, Tao and Zhang, Kaihao and Wu, Bizhu and Chen, Guanying and Liu, Wei and Luo, Wenhan},
+  journal={arXiv preprint arXiv:2403.10983},
+  year={2024}
 }
 ```
 """
@@ -654,6 +658,7 @@ def main(device, segment_type):
             inputs=[prompt, negative_prompt, reference_1, reference_2, resolution, local_prompt1, local_prompt2, seed, style, identitynet_strength_ratio, adapter_strength_ratio, condition, condition_img1, controlnet_ratio, cfg_ratio],
             outputs=[gallery, gallery1]
         )
+        gr.Markdown(article)
     demo.launch(share=True)
 
 def parse_args():