fix bugs

app.py

@@ -5,8 +5,6 @@ import os
 from src.pipeline import ObjectRemovalPipeline
 from src.utils import visualize_mask
 
-# --- ZeroGPU Compatibility Shim ---
-# Allows code to run on local CPU/GPU without crashing on 'import spaces'
 try:
     import spaces
 except ImportError:
@@ -17,11 +15,10 @@ except ImportError:
                 return func
             return decorator
 
-# Initialize pipeline (Models use lazy-loading to save memory)
+# Initialize pipeline
 pipeline = ObjectRemovalPipeline()
 
 def ensure_uint8(image):
-    """Normalize image to uint8 (0-255)"""
     if image is None: return None
     image = np.array(image)
     if image.dtype != np.uint8:
@@ -31,11 +28,9 @@ def ensure_uint8(image):
 
 @spaces.GPU(duration=120)
 def step1_detect(image, text_query):
-    """Detect objects and return candidates for user selection"""
     if image is None or not text_query:
         return [], [], "Please upload image and enter text."
     
-    # 1. Detect & Rank candidates via Pipeline
     candidates, msg = pipeline.get_candidates(image, text_query)
     
     if not candidates:
@@ -43,7 +38,6 @@ def step1_detect(image, text_query):
     
     masks = [c['mask'] for c in candidates]
     
-    # 2. Visualize masks for Gallery
     gallery_imgs = []
     for i, mask in enumerate(masks):
         viz = visualize_mask(image, mask)
@@ -54,30 +48,25 @@ def step1_detect(image, text_query):
     return masks, gallery_imgs, "Select the best match below."
 
 def on_select(evt: gr.SelectData):
-    """Capture user selection from Gallery"""
     return evt.index
 
 @spaces.GPU(duration=120)
 def step2_remove(image, masks, selected_idx, prompt, shadow_exp):
-    """Inpaint the selected mask"""
     if not masks or selected_idx is None:
         return None, "Please select an object first."
     
     target_mask = masks[selected_idx]
     
-    # 3. Inpaint with Shadow Fix logic
     result = pipeline.inpaint_selected(image, target_mask, prompt, shadow_expansion=shadow_exp)
     
     return ensure_uint8(result), "Success!"
 
-# CSS for better layout and full image visibility in Gallery
 css = """
 .gradio-container {min-height: 0px !important}
 button.gallery-item {object-fit: contain !important}
 """
 
-with gr.Blocks(title="TextEraser", css=css, theme=gr.themes.Soft()) as demo:
-    # State to hold masks between steps
+with gr.Blocks(title="TextEraser") as demo:
     mask_state = gr.State([])
     idx_state = gr.State(0) 
 
@@ -90,7 +79,6 @@ with gr.Blocks(title="TextEraser", css=css, theme=gr.themes.Soft()) as demo:
             btn_detect = gr.Button("1. Detect Objects", variant="primary")
         
         with gr.Column(scale=1):
-            # Interactive Gallery (Adaptable size)
             gallery = gr.Gallery(
                 label="Candidates (Select One)", 
                 columns=2, 
@@ -109,7 +97,6 @@ with gr.Blocks(title="TextEraser", css=css, theme=gr.themes.Soft()) as demo:
         with gr.Column(scale=1):
             output_image = gr.Image(label="Final Result", height=400)
 
-    # Event Wiring
     btn_detect.click(
         fn=step1_detect,
         inputs=[input_image, text_query],
@@ -126,8 +113,7 @@ with gr.Blocks(title="TextEraser", css=css, theme=gr.themes.Soft()) as demo:
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument("--share", action="store_true", help="Create a public link (Colab)")
+    parser.add_argument("--share", action="store_true")
     args = parser.parse_args()
     
-    # queue() is required for ZeroGPU
-    demo.queue().launch(share=args.share)
+    demo.queue().launch(share=args.share, css=css)

requirements.txt

@@ -19,4 +19,6 @@ Pillow
 sniffio
 spaces
 clip
-git+https://github.com/facebookresearch/sam2.git
+git+https://github.com/facebookresearch/sam2.git
+accelerate
+ftfy