Display demo description

app.py

@@ -101,19 +101,49 @@ def run_inference(prompt: str):
 
 if __name__ == "__main__":
 
-    # create UI
-    demo = gr.Interface(
-        fn=run_inference,
-        inputs="text",
-        outputs=gr.Plot(),
-        title="SALAD: Text-Guided Shape Generation",
-        description="Describe a chair",
-        examples=[
-            "an office chair",
-            "a chair with armrests",
-            "a chair without armrests",
-        ]
-    )
-    # initiate
+    title = "SALAD: Text-Guided Shape Generation"
+
+    description_text = '''
+    This demo features text-guided 3D shape generation from our work <a href="https://arxiv.org/abs/2303.12236">SALAD: Part-Level Latent Diffusion for 3D Shape Generation and Manipulation, ICCV 2023</a>.  
+    Please refer to our <a href="https://salad3d.github.io/">project page</a> for details.
+    '''
+
+    # create UI        
+    with gr.Blocks(title=title) as demo:
+
+        # description of demo
+        gr.Markdown(description_text)
+
+        # inputs
+        with gr.Row():
+            prompt_textbox = gr.Textbox(placeholder="Describe a chair.")
+
+            with gr.Row():
+                run_button = gr.Button(value="Generate")
+                clear_button = gr.ClearButton(
+                    value="Clear",
+                    components=[prompt_textbox],
+                )
+
+        # display examples
+        examples = gr.Examples(
+            examples=[
+                "an office chair",
+                "a chair with armrests",
+                "a chair without armrests",
+            ],
+            inputs=[prompt_textbox],
+        )
+
+        # outputs
+        mesh_viewport = gr.Plot()
+
+        # run inference
+        run_button.click(
+            run_inference,
+            inputs=[prompt_textbox],
+            outputs=[mesh_viewport],
+        )
+
     demo.queue(max_size=30)
-    demo.launch()
+    demo.launch()