Update app.py

app.py

@@ -389,7 +389,25 @@ def do_inference(input_3d, sample_seed=0, do_sampling=False, do_marching_cubes=F
     artist_mesh.export(save_name)
     output_render = wireframe_render(artist_mesh)
     return input_save_name, input_render_res, save_name, output_render
-    
+
+# Output gradio
+output_model_obj = gr.Model3D(
+    label="Generated Mesh (OBJ Format)",
+    display_mode="wireframe",
+    clear_color=[1, 1, 1, 1],
+)
+preprocess_model_obj = gr.Model3D(
+    label="Processed Input Mesh (OBJ Format)",
+    display_mode="wireframe",
+    clear_color=[1, 1, 1, 1],
+)
+input_image_render = gr.Image(
+    label="Wireframe Render of Processed Input Mesh",
+)
+output_image_render = gr.Image(
+    label="Wireframe Render of Generated Mesh",
+)
+
 ###############################################################################
 # Gradio
 ###############################################################################