Update app.py

app.py

@@ -1,7 +1,6 @@
 import shlex
 import subprocess
 
-
 subprocess.run(shlex.split("pip install pip==24.0"), check=True)
 subprocess.run(
     shlex.split(
@@ -14,12 +13,9 @@ subprocess.run(
     ), check=True
 )
 
-
 if __name__ == "__main__":
     from huggingface_hub import snapshot_download
-
     snapshot_download("public-data/Unique3D", repo_type="model", local_dir="./ckpt")
-
     import os
     import sys
     sys.path.append(os.curdir)
@@ -27,13 +23,11 @@ if __name__ == "__main__":
     torch.set_float32_matmul_precision('medium')
     torch.backends.cuda.matmul.allow_tf32 = True
     torch.set_grad_enabled(False)
-
 import fire
 import gradio as gr
 from gradio_app.gradio_3dgen import create_ui as create_3d_ui
 from gradio_app.all_models import model_zoo
 
-
 _TITLE = '''Unique3D: High-Quality and Efficient 3D Mesh Generation from a Single Image'''
 _DESCRIPTION = '''
 <div>
@@ -45,6 +39,16 @@ _DESCRIPTION = '''
 * High-fidelity and diverse textured meshes generated by Unique3D from single-view images.
 * The demo is still under construction, and more features are expected to be implemented soon.
 * If the Huggingface Demo is overcrowded or fails to produce stable results, you can use the Online Demo [aiuni.ai](https://www.aiuni.ai/), which is free to try (get the registration invitation code Join Discord: https://discord.gg/aiuni). However, the Online Demo is slightly different from the Gradio Demo, in that the inference speed is slower, but the generation is much more stable.
+
+<div style="display: flex; flex-direction: column; justify-content: center; align-items: center; text-align: center; margin-top: 20px;">
+    <p style="display: flex; gap: 6px;">
+        <a href="https://huggingface.co/spaces/cavargas10/Unico3D?duplicate=true">
+            <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-lg.svg" alt="Duplicate this Space">
+        </a>
+    </p>
+    <p>Click the button above to duplicate this space and enjoy faster inference on the GPU of your choice.</p>
+    <p style="margin-top: 10px;">Special thanks to <strong>@hysts</strong> for helping to make the "Duplicate this Space" functionality work correctly!</p>
+</div>
 '''
 
 def launch():
@@ -59,7 +63,6 @@ def launch():
                 gr.Markdown('# ' + _TITLE)
         gr.Markdown(_DESCRIPTION)
         create_3d_ui("wkl")
-
     demo.queue().launch(share=True)
     
 if __name__ == '__main__':