Update app.py

app.py

@@ -16,10 +16,8 @@ def segment_everything(image):
     segmentation = (preds.numpy() * 255).astype(np.uint8)
     return Image.fromarray(segmentation)
 
-def segment_box(image, box):
-    if box is None:
-        return image
-    x1, y1, x2, y2 = box
+def segment_box(image, x1, y1, x2, y2):
+    x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
     cropped_image = image[y1:y2, x1:x2]
     
     inputs = processor(text=["object"], images=[cropped_image], padding="max_length", return_tensors="pt")
@@ -43,7 +41,12 @@ with gr.Blocks() as demo:
     gr.Markdown("# Segment Anything-like Demo")
     with gr.Row():
         with gr.Column(scale=1):
-            input_image = gr.Image(label="Input Image", tool="select")
+            input_image = gr.Image(label="Input Image")
+            with gr.Row():
+                x1_input = gr.Number(label="X1")
+                y1_input = gr.Number(label="Y1")
+                x2_input = gr.Number(label="X2")
+                y2_input = gr.Number(label="Y2")
             with gr.Row():
                 everything_btn = gr.Button("Everything")
                 box_btn = gr.Button("Box")
@@ -58,7 +61,7 @@ with gr.Blocks() as demo:
     
     box_btn.click(
         fn=segment_box,
-        inputs=[input_image, input_image],
+        inputs=[input_image, x1_input, y1_input, x2_input, y2_input],
         outputs=[output_image]
     )