code push

app.py

@@ -16,10 +16,6 @@ title = "<center><strong><font size='10'> Fast Segment Anything </font></strong>
 
 css = "h1 { text-align: center } .about { text-align: justify; padding-left: 10%; padding-right: 10%; }"
 
-examples = [["examples/sa_8776.jpg"], ["examples/sa_414.jpg"], ["examples/sa_1309.jpg"], ["examples/sa_11025.jpg"],
-            ["examples/sa_561.jpg"], ["examples/sa_192.jpg"], ["examples/sa_10039.jpg"], ["examples/sa_862.jpg"]]
-
-default_example = examples[0]
 
 def segment_everything(
     input,
@@ -130,12 +126,7 @@ def get_points_with_draw(image, label, evt: gr.SelectData):
     return image
 
 
-cond_img_e = gr.Image(label="Input", value=default_example[0], type='pil')
-cond_img_p = gr.Image(label="Input with points", value=default_example[0], type='pil')
 cond_img_t = gr.Image(label="Input with text", value="examples/dogs.jpg", type='pil')
-
-segm_img_e = gr.Image(label="Segmented Image", interactive=False, type='pil')
-segm_img_p = gr.Image(label="Segmented Image with points", interactive=False, type='pil')
 segm_img_t = gr.Image(label="Segmented Image with text", interactive=False, type='pil')
 
 global_points = []
@@ -146,7 +137,7 @@ input_size_slider = gr.components.Slider(minimum=512,
                                          value=1024,
                                          step=64,
                                          label='Input_size',
-                                         info='Our model was trained on a size of 1024')
+                                         info='The model was trained on a size of 1024')
 
 with gr.Blocks(css=css, title='Fast Segment Anything') as demo:
     with gr.Row():
@@ -182,9 +173,9 @@ with gr.Blocks(css=css, title='Fast Segment Anything') as demo:
                         clear_btn_t = gr.Button("Clear", variant="secondary")
 
                 gr.Markdown("Try some of the examples below ⬇️")
-                gr.Examples(examples=[["examples/dogs.jpg"], ["examples/fruits.jpg"], ["examples/flowers.jpg"]],
+                gr.Examples(examples=[["examples/dogs.jpg"], ["examples/fruits.jpg"], ["examples/flowers.jpg"], ["examples/dog2.jpg"], ["examples/cat.jpg"], ["examples/person.jpg"]],
                             inputs=[cond_img_t],
-                            examples_per_page=4)
+                            examples_per_page=6)
 
             with gr.Column():
                 with gr.Accordion("Advanced options", open=False):
@@ -216,7 +207,7 @@ with gr.Blocks(css=css, title='Fast Segment Anything') as demo:
         return None, None, None
 
     # clear_btn_p.click(clear, outputs=[cond_img_p, segm_img_p])
-    clear_btn_t.click(clear_text, outputs=[cond_img_p, segm_img_p, text_box])
+    clear_btn_t.click(clear_text, outputs=[text_box])
 
 demo.queue()
 demo.launch()