Update app.py

app.py

@@ -158,6 +158,10 @@ def load_pil_img():
     from PIL import Image
     return Image.open("example_tmp/text/out_text_0.png")
 
+def change_image(impath):
+    return gr.Image(value = impath, type="numpy", height=LENGTH, width=LENGTH, interactive=True)
+                
+
 import shutil
 if os.path.isdir("./example_tmp"):
     shutil.rmtree("./example_tmp")
@@ -186,9 +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.\n
-                Afterwards, you can modify the mask ID and prompt for image editing. \n
-                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.\nAfterwards, you can modify the mask ID and prompt for image editing. \nThe D-edit link is: https://arxiv.org/abs/2403.04880v2"</p>""")
 
             with gr.Column():
                 result_info0 = gr.Text(label="Response")
@@ -312,6 +314,11 @@ with gr.Blocks() as demo:
                         gr.Info('Image editing completed.')
                         return load_pil_img()
 
+                    example_inps = [['./img.png'],['./img2.png']]
+                    gr.Examples(examples=example_inps, inputs=[img_path],
+                            label='examples', cache_examples='lazy', outputs=[output_image],
+                            fn=change_image)
+
 
         canvas.upload(image_change, inputs=[], outputs=[slider,add_button2])            
         add_button.click(run_optimization_wrapper,