add example imgs

app.py

@@ -190,7 +190,7 @@ with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
                 canvas = gr.Image(value = "./img.png", type="numpy",  label="Draw Mask", show_label=True, height=LENGTH, width=LENGTH, interactive=True)
-                gr.Markdown("""<p style="text-align: left; font-size: 16px">Each image requires a single segmentation and optimization operation.<br>Afterwards, you can modify the mask ID and prompt for image editing. <br>The D-edit link is: https://arxiv.org/abs/2403.04880v2"</p>""")
+                gr.Markdown("""<p style="text-align: left; font-size: 16px">Each image requires a single segmentation and optimization operation.<br>Afterwards, you can modify the mask ID and prompt for image editing. <br>The link of D-edit paper: https://arxiv.org/abs/2403.04880v2</p>""")
 
             with gr.Column():
                 result_info0 = gr.Text(label="Response")
@@ -199,7 +199,7 @@ with gr.Blocks() as demo:
 
                 # mask_np_list_updated.value = copy.deepcopy(mask_np_list.value) #!!
                 mask_np_list_updated = mask_np_list
-                gr.Markdown("""<p style="text-align: center; font-size: 20px">Edit Mask (Optional)</p>""")
+                gr.Markdown("""<p style="text-align: center; font-size: 20px">Edit Mask (Do not change it during the optimization)</p>""")
                 slider =  gr.Slider(0, 20, step=1, label = 'mask id',  visible=False)
                 label = gr.Text(label='label')
 
@@ -211,7 +211,7 @@ with gr.Blocks() as demo:
                 result_info = gr.Text(label="Response")
                 
                 opt_flag = gr.State(0)
-                gr.Markdown("""<p style="text-align: center; font-size: 20px">Optimization settings (SD)</p>""")
+                gr.Markdown("""<p style="text-align: center; font-size: 20px">Optimization settings</p>""")
                 with gr.Accordion(label="Advanced settings", open=False):
                     num_tokens = gr.Number(value="5", label="num tokens to represent each object", interactive= True)
                     num_tokens_global = num_tokens
@@ -271,7 +271,7 @@ with gr.Blocks() as demo:
                     # canvas_text_edit = gr.Gallery(label = "Edited results")
                 
                 with gr.Column():
-                    gr.Markdown("""<p style="text-align: center; font-size: 20px">Editing setting (SD)</p>""")
+                    gr.Markdown("""<p style="text-align: center; font-size: 20px">Editing setting</p>""")
                     tgt_prompt =  gr.Textbox(value="text prompt", label="Editing: Text prompt", interactive= True )
                     with gr.Accordion(label="Advanced settings", open=False):
                         slider2 = gr.Slider(0, 20, step=1, label = 'mask id',  visible=False)
@@ -314,7 +314,7 @@ with gr.Blocks() as demo:
                         gr.Info('Image editing completed.')
                         return load_pil_img()
 
-                    example_inps = [['./img.png'],['./img2.png'],['./img3.png']]
+                    example_inps = [['./img.png'],['./img2.png'],['./img3.png'],['./img4.png']]
                     gr.Examples(examples=example_inps, inputs=[canvas],
                             label='examples', cache_examples='lazy', outputs=[],
                             fn=change_image)