Update gradio_demo.py

gradio_demo.py

@@ -1174,8 +1174,7 @@ with block:
     with gr.Tab("Background", visible=True):
         # empty cache
     
-        mask_mover = MaskMover()
-        
+        mask_mover = gr.State(MaskMover())
         # with torch.no_grad():
         #     # Update the input channels to 12
         #     new_conv_in = torch.nn.Conv2d(12, unet.conv_in.out_channels, unet.conv_in.kernel_size, unet.conv_in.stride, unet.conv_in.padding)  # Changed from 8 to 12
@@ -1358,7 +1357,8 @@ with block:
                 return composite
     
         # Create an instance of BackgroundManager
-        bg_manager = BackgroundManager()
+        # bg_manager = BackgroundManager()
+        bg_manager = gr.State(BackgroundManager())
 
         def update_position_wrapper(background, x_pos, y_pos, scale):
             return bg_manager.update_position(background, x_pos, y_pos, scale)