Update app.py

app.py

@@ -19,7 +19,8 @@ def mask_to_rgb(mask):
     bg_transparent[mask == 1] = [0, 255, 0, 127]
     return bg_transparent
 
-def get_processed_inputs(image, points):
+def get_processed_inputs(image, points_str):
+    points = [list(map(int, point.split(','))) for point in points_str.split()]
     input_points = [points]
     inputs = processor(image, input_points=input_points, return_tensors="pt").to(device)
     with torch.no_grad():
@@ -59,7 +60,7 @@ iface = gr.Interface(
     fn=gradio_interface,
     inputs=[
         gr.Image(type="numpy", label="Input Image"),
-        gr.Image(type="numpy", label="Click to select points", tool="sketch", brush_radius=1),
+        gr.Textbox(label="Points (format: x1,y1 x2,y2 ...)", placeholder="e.g., 100,100 200,200"),
         gr.Textbox(label="Positive Prompt", placeholder="Enter positive prompt here"),
         gr.Textbox(label="Negative Prompt", placeholder="Enter negative prompt here")
     ],
@@ -68,7 +69,7 @@ iface = gr.Interface(
         gr.Image(label="Segmentation Mask")
     ],
     title="Interactive Image Inpainting",
-    description="Click on the image to select points for segmentation, provide prompts, and see the inpainted result."
+    description="Enter points as 'x1,y1 x2,y2 ...' for segmentation, provide prompts, and see the inpainted result."
 )
 
 iface.launch(share=True)