Update app.py

app.py

@@ -146,7 +146,7 @@ def image_to_3d(
     outputs = pipeline.run(
         image[1],
         seed=seed,
-        formats=["gaussian", "mesh"],
+        formats=["mesh"],
         preprocess_image=False,
         sparse_structure_sampler_params={
             "steps": ss_sampling_steps,
@@ -157,9 +157,7 @@ def image_to_3d(
             "cfg_strength": slat_guidance_strength,
         },
     )
-    video = render_utils.render_video(outputs['gaussian'][0], num_frames=120)['color']
-    video_geo = render_utils.render_video(outputs['mesh'][0], num_frames=120)['normal']
-    video = [np.concatenate([video[i], video_geo[i]], axis=1) for i in range(len(video))]
+    video = render_utils.render_video(outputs['mesh'][0], num_frames=120)['normal']
     video_path = os.path.join(user_dir, 'sample.mp4')
     imageio.mimsave(video_path, video, fps=15)
     state = pack_state(outputs['gaussian'][0], outputs['mesh'][0])
@@ -181,11 +179,15 @@ def extract_glb(
     torch.cuda.empty_cache()
     return glb_path, glb_path
 
+def reset_do_preprocess():
+    return True
+
 with gr.Blocks(delete_cache=(600, 600)) as demo:
     gr.Markdown("""
     ## Sketch to 3D with TRELLIS
     1. Fast sketch to image with SDXL Flash, using [@xinsir](https://huggingface.co/xinsir) [scribble sdxl controlnet](https://huggingface.co/xinsir/controlnet-scribble-sdxl-1.0) and [sdxl flash](https://huggingface.co/sd-community/sdxl-flash)
     2. Scalable and versatile image to 3D generation using [TRELLIS](https://trellis3d.github.io/)
+    ### ð   ¨ð    ï¸   draw or upload a sketch and click "Generate" to create a 3D asset â  ¨
     """)
     with gr.Row():
         with gr.Column():
@@ -238,14 +240,14 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
             with gr.Row():
                 extract_glb_btn = gr.Button("Extract GLB", interactive=False)
             gr.Markdown("""
-                        *NOTE: Gaussian file can be very large (~50MB), it will take a while to display and download.*
+                        *NOTE: GLB file can be downloaded after extraction.*
                         """)
         with gr.Column():
             video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
             image_prompt_processed = ImageSlider(label="processed sketch", interactive=False, type="pil", height=512)
             model_output = LitModel3D(label="Extracted GLB", exposure=10.0, height=300)
             with gr.Row():
-                download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
+                download_glb = gr.DownloadButton(label="Download GLB", interactive=False)  
 
     output_buf = gr.State()
     demo.load(start_session)
@@ -266,36 +268,31 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
     )
 
     generate_btn.click(
-    get_seed,
-    inputs=[randomize_seed, seed],
-    outputs=[seed],
+        get_seed,
+        inputs=[randomize_seed, seed],
+        outputs=[seed],
     ).then(
         image_to_3d,
         inputs=[image_prompt_processed, seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps],
         outputs=[output_buf, video_output],
     ).then(
-        # Cambiamos esto: ahora solo actualizamos el botón de GLB
-        lambda: gr.Button.update(interactive=True),
+        lambda: gr.Button(interactive=True),
         outputs=[extract_glb_btn],
     )
-    
-    # Corregimos también el evento de limpieza del video
     video_output.clear(
-        lambda: gr.Button.update(interactive=False),
+        lambda: gr.Button(interactive=False),
         outputs=[extract_glb_btn],
     )
-    
     extract_glb_btn.click(
         extract_glb,
         inputs=[output_buf, mesh_simplify, texture_size],
         outputs=[model_output, download_glb],
     ).then(
-        lambda: gr.Button.update(interactive=True),
+        lambda: gr.Button(interactive=True),
         outputs=[download_glb],
     )
-    
     model_output.clear(
-        lambda: gr.Button.update(interactive=False),
+        lambda: gr.Button(interactive=False),
         outputs=[download_glb],
     )