Update app.py

app.py

@@ -288,45 +288,22 @@ def run_grounded_sam(input_image, text_prompt, task_type, inpaint_prompt, box_th
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser("Grounded SAM demo", add_help=True)
-    parser.add_argument("--debug", action="store_true",
-                        help="using debug mode")
-    parser.add_argument("--share", action="store_true", help="share the app")
-    parser.add_argument('--no-gradio-queue', action="store_true",
-                        help='path to the SAM checkpoint')
+    parser.add_argument("--debug", action="store_true", help="using debug mode")
     args = parser.parse_args()
 
-    print(args)
-
-    block = gr.Blocks()
-    if not args.no_gradio_queue:
-        block = block.queue()
-
-    with block:
-        with gr.Row():
-            with gr.Column():
-                input_image = gr.Image(
-                    source='upload', type="pil", value="demo1.jpg")
-                task_type = gr.Dropdown(
-                    ["det", "seg", "inpainting", "automatic"], value="seg", label="task_type")
-                text_prompt = gr.Textbox(label="Text Prompt", placeholder="bear . beach .")
-                inpaint_prompt = gr.Textbox(label="Inpaint Prompt", placeholder="A dinosaur, detailed, 4K.")
-                run_button = gr.Button(label="Run")
-                with gr.Accordion("Advanced options", open=False):
-                    box_threshold = gr.Slider(
-                        label="Box Threshold", minimum=0.0, maximum=1.0, value=0.3, step=0.001
-                    )
-                    text_threshold = gr.Slider(
-                        label="Text Threshold", minimum=0.0, maximum=1.0, value=0.25, step=0.001
-                    )
-                    iou_threshold = gr.Slider(
-                        label="IOU Threshold", minimum=0.0, maximum=1.0, value=0.8, step=0.001
-                    )
-                    inpaint_mode = gr.Dropdown(
-                        ["merge", "first"], value="merge", label="inpaint_mode")
-
-            with gr.Column():
-                gallery = gr.Gallery( 
-                    show_label=False, elem_id="gallery"
-                ).style(preview=True, grid=2, object_fit="scale-down")
-
-        block.launch(debug=args.debug, share=args.share, show_error=True)
+    inputs = [
+        gr.inputs.Image(type="pil", label="Input Image"),
+        gr.inputs.Dropdown(["det", "seg", "inpainting", "automatic"], label="Task Type"),
+        gr.inputs.Textbox(label="Text Prompt", placeholder="bear . beach ."),
+        gr.inputs.Textbox(label="Inpaint Prompt", placeholder="A dinosaur, detailed, 4K."),
+        gr.inputs.Slider(label="Box Threshold", minimum=0.0, maximum=1.0, value=0.3, step=0.001),
+        gr.inputs.Slider(label="Text Threshold", minimum=0.0, maximum=1.0, value=0.25, step=0.001),
+        gr.inputs.Slider(label="IOU Threshold", minimum=0.0, maximum=1.0, value=0.8, step=0.001),
+        gr.inputs.Dropdown(["merge", "first"], label="Inpaint Mode")
+    ]
+
+    outputs = gr.outputs.Image(type="pil", label="Output Image")
+
+    iface = gr.Interface(run_demo, inputs, outputs, title="Grounded SAM Demo")
+
+    iface.launch(debug=args.debug)