Update app.py

app.py

@@ -139,7 +139,7 @@ def process(input_cond, input_num_steps, input_seed=42, input_cfg=6.0):
         recon_param = torch.concat([recon_srt_param, recon_feat_param], dim=-1)
         visualize_video_primvolume(config.output_dir, batch, recon_param, 15, rm, device)
         prim_params = {'srt_param': recon_srt_param[0].detach().cpu(), 'feat_param': recon_feat_param[0].detach().cpu()}
-    return output_rgb_video_path, output_prim_video_path, output_mat_video_path, gr.update(interactive=True), prim_params
+    return output_rgb_video_path, output_prim_video_path, output_mat_video_path, prim_params
 
 def export_mesh(prim_params, uv_unwrap="Faster", remesh="No", mc_resolution=256, decimate=100000):
     # exporting GLB mesh
@@ -247,7 +247,7 @@ with block:
                     )
 
     input_image.change(background_remove_process, inputs=[input_image], outputs=[button_gen, current_fg_state, removal_previewer])
-    button_gen.click(process, inputs=[current_fg_state, input_num_steps, input_seed, input_cfg], outputs=[output_rgb_video, output_prim_video, output_mat_video, export_glb_btn, prim_param_state])
+    button_gen.click(process, inputs=[current_fg_state, input_num_steps, input_seed, input_cfg], outputs=[output_rgb_video, output_prim_video, output_mat_video, prim_param_state])
     prim_param_state.change(export_mesh, inputs=[prim_param_state, input_unwrap, input_remesh, input_mc_resolution], outputs=[output_glb, hdr_row])
     
     gr.Examples(
@@ -256,7 +256,7 @@ with block:
             for f in os.listdir("assets/examples")
         ],
         inputs=[input_image],
-        outputs=[output_rgb_video, output_prim_video, output_mat_video, export_glb_btn],
+        outputs=[output_rgb_video, output_prim_video, output_mat_video, prim_param_state],
         fn=lambda x: process(input_image=x),
         cache_examples=False,
         label='Single Image to 3D PBR Asset'