Update app.py

app.py

@@ -1,9 +1,35 @@
 import gradio as gr
 import random
 
+_TITLE = '''
+# Direct3D-S2 Gigascale 3D Generation Made Easy with Spatial Sparse Attention
+'''
+
+_DESCRIPTION = '''
+### [GitHub](https://github.com/DreamTechAI/Direct3D-S2) | [Arxiv](https://arxiv.org/pdf/2505.17412) | [Project Page](https://nju-3dv.github.io/projects/Direct3D-S2/)
+'''
+
+_CITE_ = r"""
+---
+📝 **Citation**
+If you find our work useful for your research or applications, please cite using this bibtex:
+```bibtex
+@article{wu2025direct3ds2gigascale3dgeneration,
+  title={Direct3D-S2: Gigascale 3D Generation Made Easy with Spatial Sparse Attention}, 
+  author={Shuang Wu and Youtian Lin and Feihu Zhang and Yifei Zeng and Yikang Yang and Yajie Bao and Jiachen Qian and Siyu Zhu and Philip Torr and Xun Cao and Yao Yao},
+  journal={arXiv preprint arXiv:2505.17412},
+  year={2025}
+}
+```
+📧 **Contact**
+If you have any questions, feel free to contact us at <b>wushuang9811@outlook.com</b>.
+"""
+
 src_list = ['https://api.zmkj.site/', 'https://api1.zmkj.site/', 'https://api2.zmkj.site/']
 
 with gr.Blocks() as demo:
+    gr.Markdown(_TITLE)
+    gr.Markdown(_DESCRIPTION)
     gr.HTML(f"""
     <iframe 
         src="{random.choice(src_list)}" 
@@ -13,5 +39,6 @@ with gr.Blocks() as demo:
         allow="fullscreen"
     ></iframe>
     """)
+    gr.Markdown(_CITE_)
 
 demo.launch()