update

app.py

@@ -126,7 +126,7 @@ def sample(image, seed, category):
     command = f"python ./scripts/generate.py --config ./configs/demo/{category}_demo.yaml --output_path {mesh_fpath} --seed {seed} --params_path {params_fpath}"
     os.system(command)
     
-    return mesh_fpath
+    return mesh_fpath, mesh_fpath
 
 
 import gradio as gr
@@ -219,20 +219,20 @@ with gr.Blocks() as demo:
                     [os.path.join(f"examples/chair", img_name) for img_name in sorted(os.listdir(f"examples/chair"))],
                     inputs=[input_image],
                     label="Examples",
-                    examples_per_page=10
+                    examples_per_page=7
                 )
                 generator_category.change(update_examples, generator_category, outputs=examples.dataset)
 
         with gr.Column():
             with gr.Row():
                 with gr.Tab("Geometry"):
-                    output_model_obj = gr.Model3D(
+                    output_model_wireframe = gr.Model3D(
                         label="Output Model (Wireframe)",
                         #width=768,
                         display_mode="wireframe",
                         interactive=False
                     )
-                    output_model_obj = gr.Model3D(
+                    output_model_solid = gr.Model3D(
                         label="Output Model (Solid)",
                         #width=768,
                         interactive=False,
@@ -249,7 +249,7 @@ with gr.Blocks() as demo:
     ).success(
         fn=sample,
         inputs=[processed_image, sample_seed, generator_category],
-        outputs=[output_model_obj],
+        outputs=[output_model_wireframe, output_model_solid],
     )
 
 demo.queue()