Add smooth shading

app.py

@@ -21,6 +21,9 @@ from mpl_toolkits.mplot3d.art3d import Poly3DCollection
 from PIL import Image
 import io
 
+# For smooth shading
+from trimesh.smoothing import smooth_shade
+
 model = load_v2()
 
 device = torch.device('cuda')
@@ -168,6 +171,9 @@ def do_inference(input_3d, sample_seed=0, do_sampling=False, do_marching_cubes=F
     artist_mesh.remove_unreferenced_vertices()
     artist_mesh.fix_normals()
 
+     if do_smooth_shading:
+        smooth_shade(artist_mesh)
+
     if artist_mesh.visual.vertex_colors is not None:
         orange_color = np.array([255, 165, 0, 255], dtype=np.uint8)
 
@@ -239,6 +245,7 @@ with (gr.Blocks() as demo):
             with gr.Row():
                 with gr.Group():
                     do_marching_cubes = gr.Checkbox(label="Preprocess with Marching Cubes", value=False)
+                    do_smooth_shading = gr.Checkbox(label="Apply Smooth Shading", value=False)
                     do_sampling = gr.Checkbox(label="Random Sampling", value=False)
                     sample_seed = gr.Number(value=0, label="Seed Value", precision=0)
 
@@ -274,7 +281,7 @@ with (gr.Blocks() as demo):
 
     submit.click(
         fn=do_inference,
-        inputs=[input_3d, sample_seed, do_sampling, do_marching_cubes],
+        inputs=[input_3d, sample_seed, do_sampling, do_marching_cubes, do_smooth_shading],
         outputs=[preprocess_model_obj, input_image_render, output_model_obj, output_image_render],
     )