ui

app.py

@@ -267,30 +267,26 @@ def reset_feat2gs_state():
 
 _TITLE = '''Feat2GS Demo'''
 _DESCRIPTION = '''
-<div style="display: flex; justify-content: center; align-items: center;">
-    <div style="width: 100%; text-align: center; font-size: 30px;">
-        <strong><span style="font-family: 'Comic Sans MS';"><span style="color: #E0933F">Feat</span><span style="color: #B24C33">2</span><span style="color: #E0933F">GS</span></span>: Probing Visual Foundation Models with Gaussian Splatting</strong>
-    </div>
-</div> 
-<p></p>
-<div align="center">
-    <a style="display:inline-block" href="https://fanegg.github.io/Feat2GS/"><img src='https://img.shields.io/badge/Project-Website-green.svg'></a>&nbsp;
-    <a style="display:inline-block" href="https://arxiv.org/abs/2412.09606"><img src="https://img.shields.io/badge/Arxiv-2412.09606-b31b1b.svg?logo=arXiv" alt='arxiv'></a>&nbsp;
-    <a style="display:inline-block" href="https://youtu.be/4fT5lzcAJqo?si=_fCSIuXNBSmov2VA"><img src='https://img.shields.io/badge/Video-E33122?logo=Youtube'></a>&nbsp;
-    <a style="display:inline-block" href="https://github.com/fanegg/Feat2GS"><img src="https://img.shields.io/badge/Code-black?logo=Github" alt='Code'></a>&nbsp;
-    <a title="X" href="https://twitter.com/faneggchen" target="_blank" rel="noopener noreferrer" style="display: inline-block;">
-        <img src="https://img.shields.io/badge/@Yue%20Chen-black?logo=X" alt="X">
-    </a>&nbsp;
-    <a title="Bluesky" href="https://bsky.app/profile/fanegg.bsky.social" target="_blank" rel="noopener noreferrer" style="display: inline-block;">
-        <img src="https://img.shields.io/badge/@Yue%20Chen-white?logo=Bluesky" alt="Bluesky">
-    </a>
-</div>
-<p></p>
+# Feat2GS: Probing Visual Foundation Models with Gaussian Splatting
+\n\n<a style="display:inline-block" href="https://fanegg.github.io/Feat2GS/"><img src='https://img.shields.io/badge/Project-Website-green.svg'></a>&nbsp;<a style="display:inline-block" href="https://arxiv.org/abs/2412.09606"><img src="https://img.shields.io/badge/Arxiv-2412.09606-b31b1b.svg?logo=arXiv" alt='arxiv'></a>&nbsp;<a style="display:inline-block" href="https://youtu.be/4fT5lzcAJqo?si=_fCSIuXNBSmov2VA"><img src='https://img.shields.io/badge/Video-E33122?logo=Youtube'></a>&nbsp;<a style="display:inline-block" href="https://github.com/fanegg/Feat2GS"><img src="https://img.shields.io/badge/Code-black?logo=Github" alt='Code'></a>&nbsp;<a title="X" href="https://twitter.com/faneggchen" target="_blank" rel="noopener noreferrer" style="display: inline-block;"><img src="https://img.shields.io/badge/@Yue%20Chen-black?logo=X" alt="X"></a>&nbsp;<a title="Bluesky" href="https://bsky.app/profile/fanegg.bsky.social" target="_blank" rel="noopener noreferrer" style="display: inline-block;"><img src="https://img.shields.io/badge/@Yue%20Chen-white?logo=Bluesky" alt="Bluesky"></a>
 '''
 
+_CITE_ = r"""
+## 📝 **Citation**
+If you find our work useful for your research or applications, please consider citing the following paper:
+```bibtex
+@article{chen2024feat2gs,
+  title={Feat2GS: Probing Visual Foundation Models with Gaussian Splatting},
+  author={Chen, Yue and Chen, Xingyu and Chen, Anpei and Pons-Moll, Gerard and Xiu, Yuliang},
+  journal={arXiv preprint arXiv:2412.09606},
+  year={2024}
+}
+```
+"""
+
 
 # demo = gr.Blocks(title=_TITLE).queue()
-demo = gr.Blocks(css=""".gradio-container {margin: 0 !important; min-width: 100%};""", title="Feat2GS Demo").queue()
+demo = gr.Blocks(css=""".gradio-container {margin: 0 !important; min-width: 100%};""", title="Feat2GS Demo", theme=gr.themes.Origin()).queue()
 with demo:
     dust3r_state = gr.State(None)
     feat2gs_state = gr.State(None)
@@ -401,16 +397,17 @@ with demo:
     )
     render_run.click(run_render, inputs=[dust3r_state, feat2gs_state, cam_traj], outputs=[output_video])
 
-
-    gr.Examples(
-        examples=[
-            "plushies",
-        ],
-        inputs=[input_path],
-        outputs=[dust3r_model, feat_image, output_model, output_video],
-        fn=lambda x: process_example(inputfiles=None, input_path=x),
-        cache_examples=True,
-        label='Examples'
-    )
+    gr.Markdown(_CITE_)
+
+    # gr.Examples(
+    #     examples=[
+    #         "plushies",
+    #     ],
+    #     inputs=[input_path],
+    #     outputs=[dust3r_model, feat_image, output_model, output_video],
+    #     fn=lambda x: process_example(inputfiles=None, input_path=x),
+    #     cache_examples=True,
+    #     label='Examples'
+    # )
 
 demo.launch(server_name="0.0.0.0", share=False)