remove gbl extraction

app.py

@@ -487,28 +487,28 @@ def create_interface():
                         value=DEFAULT_NUM_INFERENCE_STEPS,
                     )
                 
-                # with gr.Tab("3D Generation Settings"):
-                #     trellis_seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
-                #     trellis_randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
-                #     gr.Markdown("Stage 1: Sparse Structure Generation")
-                #     with gr.Row():
-                #         ss_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance Strength", value=7.5, step=0.1)
-                #         ss_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=12, step=1)
-                #     gr.Markdown("Stage 2: Structured Latent Generation")
-                #     with gr.Row():
-                #         slat_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance Strength", value=3.0, step=0.1)
-                #         slat_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=12, step=1)
-
-                # with gr.Tab("GLB Extraction Settings"):
-                #     mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify", value=0.95, step=0.01)
-                #     texture_size = gr.Slider(512, 2048, label="Texture Size", value=1024, step=512)
+                with gr.Tab("3D Generation Settings"):
+                    trellis_seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
+                    trellis_randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
+                    gr.Markdown("Stage 1: Sparse Structure Generation")
+                    with gr.Row():
+                        ss_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance Strength", value=7.5, step=0.1)
+                        ss_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=12, step=1)
+                    gr.Markdown("Stage 2: Structured Latent Generation")
+                    with gr.Row():
+                        slat_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance Strength", value=3.0, step=0.1)
+                        slat_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=12, step=1)
+
+                with gr.Tab("GLB Extraction Settings"):
+                    mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify", value=0.95, step=0.01)
+                    texture_size = gr.Slider(512, 2048, label="Texture Size", value=1024, step=512)
             
-            # with gr.Row():
-            #     extract_glb_btn = gr.Button("Extract GLB", interactive=False)
-            #     extract_gs_btn = gr.Button("Extract Gaussian", interactive=False)
-            # gr.Markdown("""
-            #             *NOTE: Gaussian file can be very large (~50MB), it will take a while to display and download.*
-            #             """)
+            with gr.Row():
+                extract_glb_btn = gr.Button("Extract GLB", interactive=False)
+                extract_gs_btn = gr.Button("Extract Gaussian", interactive=False)
+            gr.Markdown("""
+                        *NOTE: Gaussian file can be very large (~50MB), it will take a while to display and download.*
+                        """)
 
             output_buf = gr.State()
 
@@ -534,16 +534,16 @@ def create_interface():
             outputs=[generated_image, message_box]
         )
 
-        # gr.on(
-        #     triggers=[gen3d_button.click],
-        #     fn=image_to_3d,
-        #     inputs=[generated_image, trellis_seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps],
-        #     outputs=[output_state, video_output],
-        # ).then(
-        #     # Update button states after successful 3D generation
-        #     lambda: (gr.Button.update(interactive=True), gr.Button.update(interactive=True), "3D model generated successfully"),
-        #     outputs=[extract_glb_btn, extract_gs_btn, message_box]
-        # )
+        gr.on(
+            triggers=[gen3d_button.click],
+            fn=image_to_3d,
+            inputs=[generated_image, trellis_seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps],
+            outputs=[output_state, video_output],
+        ).then(
+            # Update button states after successful 3D generation
+            lambda: (gr.Button.update(interactive=True), gr.Button.update(interactive=True), "3D model generated successfully"),
+            outputs=[extract_glb_btn, extract_gs_btn, message_box]
+        )
         
         # # Add handlers for GLB and Gaussian extraction
         # gr.on(