update

app.py

@@ -64,7 +64,7 @@ def end_session(req: gr.Request) -> None:
 
 
 with gr.Blocks(
-    delete_cache=(43200, 43200), theme=Default(primary_hue=slate)
+    delete_cache=(43200, 43200)
 ) as demo:
     gr.Markdown(
         f"""
@@ -269,16 +269,17 @@ with gr.Blocks(
                 loop=True,
                 height=300,
             )
-            model_output_gs = LitModel3D(
+            model_output_gs = gr.Model3D(
                 label="Gaussian Representation", height=300, interactive=False
             )
             aligned_gs = gr.Textbox(visible=False)
-            # with gr.Row():
-            #     model_output_mesh = gr.Model3D(
-            #         label="Mesh Representation",
-            #         height=300,
-            #         interactive=False
-            #     )
+            with gr.Row():
+                model_output_mesh = LitModel3D(
+                    label="Mesh Representation",
+                    height=300,
+                    exposure=10,
+                    interactive=False
+                )
             gr.Markdown(
                 """ The rendering of `Gaussian Representation` takes additional 10s. """  # noqa
             )
@@ -334,7 +335,7 @@ with gr.Blocks(
             download_urdf,
             model_output_gs,
             aligned_gs,
-            None,
+            model_output_mesh,
             video_output,
             asset_cat_text,
             height_range_text,
@@ -385,7 +386,7 @@ with gr.Blocks(
             extract_urdf_btn,
             download_urdf,
             model_output_gs,
-            None,
+            model_output_mesh,
             video_output,
             asset_cat_text,
             height_range_text,
@@ -451,7 +452,7 @@ with gr.Blocks(
             # gr.State(lambda: IMAGESR_MODEL),
         ],
         outputs=[
-            None,
+            model_output_mesh,
             model_output_gs,
             model_output_obj,
             aligned_gs,

requirements.txt

@@ -21,8 +21,8 @@ pyvista==0.36.1
 openai==1.58.1
 spconv-cu118==2.3.8
 transformers==4.42.4
-# gradio_litmodel3d==0.0.1
-gradio==5.12.0
+gradio_litmodel3d==0.0.1
+# gradio==5.12.0
 sentencepiece==0.2.0
 diffusers==0.31.0
 xatlas==0.0.9