Update

README.md

@@ -4,7 +4,7 @@ emoji: 🧢
 colorFrom: yellow
 colorTo: blue
 sdk: gradio
-sdk_version: 3.34.0
+sdk_version: 3.36.0
 python_version: 3.10.11
 app_file: app.py
 pinned: false

app.py

@@ -11,8 +11,6 @@ from model import Model
 
 DESCRIPTION = '# [Shap-E](https://github.com/openai/shap-e)'
 
-if (SPACE_ID := os.getenv('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 🥶 This demo does not work on CPU.</p>'
 
@@ -20,6 +18,9 @@ model = Model()
 
 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('SYSTEM') == 'spaces')
     with gr.Tabs():
         with gr.Tab(label='Text to 3D'):
             create_demo_text_to_3d(model)

app_text_to_3d.py

@@ -26,12 +26,12 @@ def create_demo(model: Model) -> gr.Blocks:
     with gr.Blocks() as demo:
         with gr.Box():
             with gr.Row(elem_id='prompt-container'):
-                prompt = gr.Text(
-                    label='Prompt',
-                    show_label=False,
-                    max_lines=1,
-                    placeholder='Enter your prompt').style(container=False)
-                run_button = gr.Button('Run').style(full_width=False)
+                prompt = gr.Text(label='Prompt',
+                                 show_label=False,
+                                 max_lines=1,
+                                 placeholder='Enter your prompt',
+                                 container=False)
+                run_button = gr.Button('Run', scale=0)
             result = gr.Model3D(label='Result', show_label=False)
             with gr.Accordion('Advanced options', open=False):
                 seed = gr.Slider(label='Seed',

style.css

@@ -11,3 +11,7 @@ h1 {
 #prompt-container {
   gap: 0;
 }
+
+#duplicate-button {
+  margin: auto;
+}