Update app.py

app.py

@@ -74,7 +74,6 @@ config.model.pop("latent_std")
 model_primx = load_from_config(config.model)
 # load rembg
 rembg_session = rembg.new_session()
-current_fg_state = None
 
 # background removal function
 def background_remove_process(input_image):
@@ -175,6 +174,7 @@ _DESCRIPTION = '''
 
 block = gr.Blocks(title=_TITLE).queue()
 with block:
+    current_fg_state = gr.State()
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown('# ' + _TITLE)
@@ -209,7 +209,7 @@ with block:
                     label="3D GLB Model",
                     visible=True,
                     clear_color=[0.0, 0.0, 0.0, 0.0],
-                    camera_position=(180, None, None),
+                    camera_position=(270, None, None),
                     tonemapping="aces",
                     contrast=1.0,
                     scale=1.0,
@@ -238,11 +238,9 @@ with block:
                         outputs=[output_glb],
                     )
 
-        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])
-        
-        export_glb_btn.click(export_mesh, inputs=[], outputs=[output_glb, hdr_row])
+    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])
+    export_glb_btn.click(export_mesh, inputs=[], outputs=[output_glb, hdr_row])
     
     gr.Examples(
         examples=[