Update app.py

app.py

@@ -12,12 +12,12 @@ import io
 sam_model = SamModel.from_pretrained("facebook/sam-vit-huge", torch_dtype=torch.float32).to("cpu")
 sam_processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
 
-# Initialize Inpainting pipeline on CPU
+# Initialize Inpainting pipeline on CPU with a compatible model
 inpaint_pipeline = StableDiffusionInpaintPipeline.from_pretrained(
-    "runwayml/stable-diffusion-inpainting",
+    "stabilityai/stable-diffusion-2-inpainting",
     torch_dtype=torch.float32
 ).to("cpu")
-# Removed model_cpu_offload as it's unnecessary for CPU
+# No need for model_cpu_offload on CPU
 
 def mask_to_rgba(mask):
     """
@@ -157,7 +157,13 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column():
-            image_input = gr.Image(label="Upload Image", type="pil", tool="point", interactive=True)
+            image_input = gr.Image(label="Upload Image", type="pil", interactive=True, elem_id="image")
+            points_input = gr.Points(
+                label="Select Points on the Object",
+                show_label=True,
+                source="image",  # Links Points to the Image component via elem_id
+                interactive=True
+            )
             prompt_input = gr.Textbox(label="Replacement Prompt", placeholder="e.g., a red sports car", lines=2)
             negative_prompt_input = gr.Textbox(label="Negative Prompt", placeholder="e.g., blurry, low quality", lines=2)
             seed_input = gr.Number(label="Seed", value=42)
@@ -167,17 +173,10 @@ with gr.Blocks() as demo:
             masked_output = gr.Image(label="Selected Object Mask Overlay")
             augmented_output = gr.Image(label="Augmented Image")
     
-    # Capture points selected on the image
-    points = gr.State([])
-    
-    def update_points(selected_points):
-        return selected_points
-    
-    image_input.select(update_points, inputs=image_input, outputs=points)
-    
+    # Bind the process function to the button click
     process_button.click(
         fn=process,
-        inputs=[image_input, points, prompt_input, negative_prompt_input, seed_input, guidance_scale_input],
+        inputs=[image_input, points_input, prompt_input, negative_prompt_input, seed_input, guidance_scale_input],
         outputs=[masked_output, augmented_output]
     )