Update app.py

app.py

@@ -6,6 +6,10 @@ subprocess.run(
     env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
     shell=True,
 )
+subprocess.run(
+    "pip install gradio==4.40.0",
+    shell=True,
+)
 import os
 import torch
 import trimesh
@@ -211,12 +215,12 @@ If you have any questions, feel free to open a discussion or contact us at <b>yi
 """
 output_model_obj = gr.Model3D(
     label="Generated Mesh (OBJ Format)",
-    #display_mode="wireframe",
+    display_mode="wireframe",
     clear_color=[1, 1, 1, 1],
 )
 preprocess_model_obj = gr.Model3D(
     label="Processed Input Mesh (OBJ Format)",
-    #display_mode="wireframe",
+    display_mode="wireframe",
     clear_color=[1, 1, 1, 1],
 )
 input_image_render = gr.Image(
@@ -232,7 +236,7 @@ with (gr.Blocks() as demo):
             with gr.Row():
                 input_3d = gr.Model3D(
                     label="Input Mesh",
-                    #display_mode="wireframe",
+                    display_mode="wireframe",
                     clear_color=[1,1,1,1],
                 )