Update

app.py

@@ -13,9 +13,6 @@ from model import Model
 
 DESCRIPTION = "# [UniDiffuser](https://github.com/thu-ml/unidiffuser)"
 
-SPACE_ID = os.getenv("SPACE_ID")
-if SPACE_ID is not None:
-    DESCRIPTION += f'\n<p>For faster inference without waiting in queue, you may duplicate the space and upgrade to GPU in settings. <a href="https://huggingface.co/spaces/{SPACE_ID}?duplicate=true"><img style="display: inline; margin-top: 0em; margin-bottom: 0em" src="https://bit.ly/3gLdBN6" alt="Duplicate Space" /></a></p>'
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶</p>"
 
@@ -116,6 +113,11 @@ def create_demo(mode_name: str) -> gr.Blocks:
 
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
+    gr.DuplicateButton(
+        value="Duplicate Space for private use",
+        elem_id="duplicate-button",
+        visible=os.getenv("SHOW_DUPLICATE_BUTTON") == "1",
+    )
     with gr.Tabs():
         with gr.TabItem("text2image"):
             create_demo("t2i")