update

app.py

@@ -61,7 +61,6 @@ def get_input_files(input_path):
 
 args = parse_args()
 
-# Create output directory (keeping your original variable name)
 LOG_PATH = args.log_path
 os.makedirs(LOG_PATH, exist_ok=True)
 
@@ -349,14 +348,36 @@ def process_3d_model(input_3d, dilated_offset, do_marching_cubes, postprocess_me
     # except Exception as e:
     #     return f"Error processing file: {str(e)}"
 
-# Title and reminder placeholders
-title = "3D Model Processing Demo"
-reminder = "Please upload your 3D model file and adjust parameters as needed."
+
+_HEADER_ = '''
+<h2><b>[SIGGRAPH 2025] PrimitiveAnything 🤗 Gradio Demo</b></h2>
+This is official demo for our SIGGRAPH 2025 paper <a href="">PrimitiveAnything: Human-Crafted 3D Primitive Assembly Generation with Auto-Regressive Transformer</a>.
+Code: <a href='#' target='_blank'>GitHub</a>. Paper: <a href='#' target='_blank'>ArXiv</a>.
+❗️❗️❗️**Important Notes:** For optimal results with fine structures, we automatically apply marching cubes and dilation operations (which differs from testing and evaluation). This prevents quality degradation in thin areas.
+'''
+
+_CITE_ = r"""
+If PrimitiveAnything is helpful, please help to ⭐ the <a href='https://github.com/PrimitiveAnything/PrimitiveAnything' target='_blank'>GitHub Repo</a>. Thanks! [![GitHub Stars](https://img.shields.io/github/stars/PrimitiveAnything/PrimitiveAnything?style=social)](https://github.com/PrimitiveAnything/PrimitiveAnything)
+---
+📝 **Citation**
+If you find our work useful for your research or applications, please cite using this bibtex:
+```bibtex
+@misc{ye2025primitiveanything,
+    title={PrimitiveAnything: Human-Crafted 3D Primitive Assembly Generation with Auto-Regressive Transformer}, 
+    author={Jingwen Ye and Yuze He and Yanning Zhou and Yiqin Zhu and Kaiwen Xiao and Yong-Jin Liu and Wei Yang and Xiao Han},
+    year={2025},
+    eprint={2505.XXXXX},
+    archivePrefix={arXiv},
+    primaryClass={cs.CV}
+}
+```
+📧 **Contact**
+If you have any questions, feel free to open a discussion or contact us at <b>hyz22@mails.tsinghua.edu.cn</b>.
+"""
 
 with gr.Blocks(title=title) as demo:
     # Title section
-    gr.Markdown(f"# {title}")
-    gr.Markdown(reminder)
+    gr.Markdown(_HEADER_)
     
     with gr.Row():
         with gr.Column():
@@ -386,5 +407,7 @@ with gr.Blocks(title=title) as demo:
         examples_per_page=14,
     )
 
+    gr.Markdown(_CITE_)
+
 if __name__ == "__main__":
     demo.launch(ssr_mode=False)