.

app.py

@@ -214,7 +214,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
 
         with gr.Column():
             video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
-            # model_output = LitModel3D(label="Extracted GLB/Gaussian", exposure=10.0, height=300)
+            model_output = LitModel3D(label="Extracted GLB/Gaussian", exposure=10.0, height=300)
             
             with gr.Row():
                 download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
@@ -244,28 +244,28 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         outputs=[extract_glb_btn, extract_gs_btn],
     )
 
-    # extract_glb_btn.click(
-    #     extract_glb,
-    #     inputs=[output_buf, mesh_simplify, texture_size],
-    #     outputs=[model_output, download_glb],
-    # ).then(
-    #     lambda: gr.Button(interactive=True),
-    #     outputs=[download_glb],
-    # )
+    extract_glb_btn.click(
+        extract_glb,
+        inputs=[output_buf, mesh_simplify, texture_size],
+        outputs=[model_output, download_glb],
+    ).then(
+        lambda: gr.Button(interactive=True),
+        outputs=[download_glb],
+    )
     
-    # extract_gs_btn.click(
-    #     extract_gaussian,
-    #     inputs=[output_buf],
-    #     outputs=[model_output, download_gs],
-    # ).then(
-    #     lambda: gr.Button(interactive=True),
-    #     outputs=[download_gs],
-    # )
-
-    # model_output.clear(
-    #     lambda: gr.Button(interactive=False),
-    #     outputs=[download_glb],
-    # )
+    extract_gs_btn.click(
+        extract_gaussian,
+        inputs=[output_buf],
+        outputs=[model_output, download_gs],
+    ).then(
+        lambda: gr.Button(interactive=True),
+        outputs=[download_gs],
+    )
+
+    model_output.clear(
+        lambda: gr.Button(interactive=False),
+        outputs=[download_glb],
+    )
     
 
 # Launch the Gradio app