remove litmodel3d

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import spaces
-from gradio_litmodel3d import LitModel3D
 
 import os
 import shutil
@@ -214,7 +213,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 = gr.Model3D(label="Extracted GLB/Gaussian", exposure=10.0, height=300)
             
             with gr.Row():
                 download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
@@ -244,23 +243,23 @@ 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],
-    # )
+    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),