update

app.py

@@ -18,7 +18,7 @@ from common import (
 )
 from gradio.themes import Default
 from gradio.themes.utils.colors import slate
-# from gradio_litmodel3d import LitModel3D
+from gradio_litmodel3d import LitModel3D
 from asset3d_gen.models.delight import DelightingModel
 from asset3d_gen.models.segment import RembgRemover, SAMPredictor
 from asset3d_gen.models.super_resolution import ImageRealESRGAN
@@ -269,15 +269,16 @@ with gr.Blocks(
                 loop=True,
                 height=300,
             )
-            model_output_gs = gr.Model3D(
+            model_output_gs = LitModel3D(
                 label="Gaussian Representation", height=300, interactive=False
             )
             aligned_gs = gr.Textbox(visible=False)
             with gr.Row():
-                model_output_mesh = gr.Model3D(
+                model_output_mesh = LitModel3D(
                     label="Mesh Representation",
                     height=300,
                     interactive=False,
+                    exposure=10,
                 )
             gr.Markdown(
                 """ The rendering of `Gaussian Representation` takes additional 10s. """  # noqa