Update app.py

app.py

@@ -38,6 +38,7 @@ def preprocess_images(images: List[Tuple[Image.Image, str]]) -> List[Image.Image
 def pack_state(gs: Gaussian, mesh: MeshExtractResult) -> dict:
     return {
         'gaussian': {
+            **gs.init_params,
             '_xyz': gs._xyz.cpu().numpy(),
             '_features_dc': gs._features_dc.cpu().numpy(),
             '_scaling': gs._scaling.cpu().numpy(),
@@ -192,7 +193,8 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
                     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)
                 multiimage_algo = gr.Radio(["stochastic", "multidiffusion"], label="Multi-image Algorithm", value="stochastic")
-                generate_btn = gr.Button("Generate")
+            
+            generate_btn = gr.Button("Generate")
             
             with gr.Accordion(label="GLB Extraction Settings", open=False):
                 mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify", value=0.95, step=0.01)