Update gradio_demo.py

gradio_demo.py

@@ -835,17 +835,12 @@ def compress_image(image):
     compressed_img = np.array(Image.open("compressed_image.jpg"))
     return compressed_img
 
-
+# @hydra.main(config_path="/home/user/app/configs", config_name="sam2_hiera_l")
 @spaces.GPU(duration=60)
 @torch.inference_mode
-@hydra.main(config_path="/home/user/app/configs", config_name="sam2_hiera_l")
-def process_image(cfg: DictConfig, input_image, input_text):
+def process_image(input_image, input_text):
     """Main processing function for the Gradio interface"""
 
-    print(f"Config: {cfg}")
-    print(f"Input Image: {input_image}")
-    print(f"Input Text: {input_text}")
-    
     # Initialize configs
     API_TOKEN = "9c8c865e10ec1821bea79d9fa9dc8720"
     SAM2_CHECKPOINT = "./checkpoints/sam2_hiera_large.pt"
@@ -1294,9 +1289,9 @@ with block:
             
         #     return mask_mover.create_composite(original_bg, x_pos, y_pos, scale)
 
-        # class BackgroundManager:
-        #     def __init__(self):
-        #         self.original_bg = None
+        class BackgroundManager:
+            def __init__(self):
+                self.original_bg = None
 
         #     # def update_position(self, background, x_pos, y_pos, scale):
         #     #     """Update composite when position changes"""
@@ -1313,13 +1308,18 @@ with block:
                 
         #     #     return mask_mover.create_composite(self.original_bg, x_pos, y_pos, scale)
             
-        #     def update_position(background, x_pos, y_pos, scale):
-        #         if background is None:
-        #             return None
-        #         # Restore a fresh copy of the original background
-        #         fresh_bg = bg_manager.original_bg.copy()
-        #         # Composite the foreground once
-        #         return mask_mover.create_composite(fresh_bg, float(x_pos), float(y_pos), float(scale))
+            # def update_position(background, x_pos, y_pos, scale):
+            #     if background is None:
+            #         return None
+            #     # Restore a fresh copy of the original background
+            #     fresh_bg = bg_manager.original_bg.copy()
+            #     # Composite the foreground once
+            #     return mask_mover.create_composite(fresh_bg, float(x_pos), float(y_pos), float(scale))
+
+  
+
+        # Create an instance of BackgroundManager
+        # bg_manager = BackgroundManager()
 
         def update_position(background, x_pos, y_pos, scale):
             if background is None:
@@ -1330,9 +1330,6 @@ with block:
 
 
 
-        # Create an instance of BackgroundManager
-        # bg_manager = BackgroundManager()
-
         input_bg.change(
         fn=lambda new_bg: setattr(bg_manager, 'original_bg', None) or new_bg,
         inputs=[input_bg],