Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import spaces
-from gradio_litmodel3d import LitModel3D
 import os
 import shutil
 os.environ['SPCONV_ALGO'] = 'native'
@@ -224,27 +223,9 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
                 with gr.Accordion(label="Generation Settings", open=False):
                     with gr.Tab(label="sketch-to-image generation"):
                         negative_prompt = gr.Textbox(label="Negative prompt")
-                        num_steps = gr.Slider(
-                        label="Number of steps",
-                        minimum=1,
-                        maximum=20,
-                        step=1,
-                        value=8,
-                    )
-                        guidance_scale = gr.Slider(
-                            label="Guidance scale",
-                            minimum=0.1,
-                            maximum=10.0,
-                            step=0.1,
-                            value=5,
-                        )
-                        controlnet_conditioning_scale = gr.Slider(
-                            label="controlnet conditioning scale",
-                            minimum=0.5,
-                            maximum=5.0,
-                            step=0.01,
-                            value=0.85,
-                        )
+                        num_steps = gr.Slider(1, 20, label="Number of steps", value=8, step=1)
+                        guidance_scale = gr.Slider(0.1, 10.0, label="Guidance scale", value=5, step=0.1)
+                        controlnet_conditioning_scale = gr.Slider(0.5, 5.0, label="controlnet conditioning scale", value=0.85, step=0.01)
                     with gr.Tab(label="3D generation"):
                         seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
                         randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
@@ -265,7 +246,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)
             image_prompt_processed = gr.Image(label="processed sketch", interactive=False, type="pil", height=512)
-            model_output = LitModel3D(label="Extracted GLB/Gaussian", exposure=10.0, height=300)
+            model_output = gr.Model3D(label="Extracted GLB/Gaussian", height=300)
             with gr.Row():
                 download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
                 download_gs = gr.DownloadButton(label="Download Gaussian", interactive=False)