Update app.py

app.py

@@ -21,8 +21,11 @@ def mask_to_rgb(mask):
     return bg_transparent
 
 def get_processed_inputs(image, points_str):
-    points = [list(map(int, point.split(','))) for point in points_str.split()]
-    input_points = [points]
+    # Parse the input string into a list of points
+    points = list(map(int, points_str.split(',')))
+    # Reshape the points into pairs
+    input_points = [[points[i:i+2]] for i in range(0, len(points), 2)]
+    
     inputs = processor(image, input_points=input_points, return_tensors="pt").to(device)
     with torch.no_grad():
         outputs = model(**inputs)
@@ -63,7 +66,7 @@ iface = gr.Interface(
     fn=gradio_interface,
     inputs=[
         gr.Image(type="numpy", label="Input Image"),
-        gr.Textbox(label="Points (format: x1,y1 x2,y2 ...)", placeholder="e.g., 100,100 200,200"),
+        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")
     ],
@@ -72,7 +75,7 @@ iface = gr.Interface(
         gr.Image(label="Segmentation Mask")
     ],
     title="Interactive Image Inpainting",
-    description="Enter points as 'x1,y1 x2,y2 ...' 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)