test modified interface

app.py

@@ -71,7 +71,7 @@ def overlay_masks_boxes_on_image(image, masks, boxes, labels, show_masks, show_b
     return output_image
 
 
-def detect_objects(image, prompt, show_masks, show_boxes, crop_options):
+def detect_objects(image, prompt, show_masks=True, show_boxes=True, crop_options="No crop"):
     image_source, image = load_image(image)
     predictor.set_image(image_source)
     
@@ -142,77 +142,30 @@ def detect_objects(image, prompt, show_masks, show_boxes, crop_options):
     
     return output_image_path
 
-block = gr.Blocks(css=".gradio-container {background-color: #f8f8f8; font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif}")
-
-with block:
-    gr.HTML("""
-    <style>
-        body {
-            background-color: #f5f5f5;
-            font-family: 'Roboto', sans-serif;
-            padding: 30px;
-        }
-    </style>
-    """)
-    
-    gr.HTML("<h1 style='text-align: center;'>Segment Any Image</h1>")
-    gr.HTML("<h3 style='text-align: center;'>Zero-Shot Object Detection, Segmentation and Cropping</h3>")
-    with gr.Row():
-        with gr.Column(width="auto"):
-            input_image = gr.Image(type='filepath', label="Upload Image")
-        with gr.Column(width="auto"):
-            output_image = gr.Image(type='filepath', label="Result")
-    with gr.Row():
-        with gr.Column(width="auto"):
-            object_search = gr.Textbox(
+
+app = gr.Interface(
+    detect_objects,
+    inputs=[gr.Image(type='filepath', label="Upload Image"),
+            gr.Textbox(
                 label="Object to Detect",
                 placeholder="Enter any text, comma separated if multiple objects needed",
                 show_label=True,
                 lines=1,
-            )
-        with gr.Column(width="auto"):
-            show_masks = gr.Checkbox(label="Show Masks", default=True)
-            show_boxes = gr.Checkbox(label="Show Boxes", default=True)
-        with gr.Column(width="auto"):
-            crop_options = gr.Radio(choices=["None", "Crop", "Inverse Crop"], label="Crop Options", default="None")
-    with gr.Row():
-        submit = gr.Button(value="Send", variant="secondary").style(full_width=True)
-
-    gr.Examples(
-        examples=[
-            ["images/tiger.jpeg", "animal from cat family", True, True],
-            ["images/car.jpeg", "a blue sports car", True, False],
-            ["images/bags.jpeg", "black bag next to the red bag", False, True],
-            ["images/deer.jpeg", "deer jumping and running across the road", True, True],
-            ["images/penn.jpeg", "sign board", True, False],
-        ],
-        inputs=[input_image, object_search, show_masks, show_boxes],
-    )
-    gr.HTML("""
-            <div style="text-align:center">
-                <p>Developed by <a href='https://www.linkedin.com/in/dekay/'>Github and Huggingface: Volkopat</a></p>
-                <p>Powered by <a href='https://segment-anything.com'>Segment Anything</a> and <a href='https://arxiv.org/abs/2303.05499'>Grounding DINO</a></p>
-                <p>Just upload an image and enter the objects to detect, segment, crop, etc.  That's all folks!</p>
-                <p>What's Zero-Shot? It means you can detect objects without any training samples!</p>
-                <p>This project is for demonstration purposes. Credits for State of the Art models go to Meta AI and IDEA Research.</p>
-            </div>
-            <style>
-                p {
-                    margin-bottom: 10px;
-                    font-size: 16px;
-                }
-                a {
-                    color: #3867d6;
-                    text-decoration: none;
-                }
-                a:hover {
-                    text-decoration: underline;
-                }
-            </style>
-            """)
-
-    submit.click(fn=detect_objects,
-                inputs=[input_image, object_search, show_masks, show_boxes, crop_options],
-                outputs=[output_image])
-
-block.launch(width=800)
+            )],
+    outputs=[
+        gr.JSON(label="Output JSON"),
+        gr.Gallery(label="Result"),
+    ],
+    examples=[
+        ["images/tiger.jpeg", "animal from cat family"],
+        ["images/car.jpeg", "a blue sports car"],
+        ["images/bags.jpeg", "black bag next to the red bag"],
+        ["images/deer.jpeg", "deer jumping and running across the road"],
+        ["images/penn.jpeg", "sign board"]
+    ],
+    title="Segment Anything",
+    description="Zero-Shot Object Detection, Segmentation and Cropping",
+    article="https://segment-anything.com",
+)
+
+app.launch()