modified: app.py

app.py

@@ -463,7 +463,7 @@ def on_upload(path, uploaded: gr.EventData):
 
 def on_change(init_image, changed: gr.EventData):
     img_path = list(changed.target.temp_files)
-    return gr.Gallery(value=[(img_path[0], "")], label="History Image", interactive=True, columns=3)
+    return gr.Gallery(value=[(img_path[0], "")], label="History Image", interactive=True, columns=3), img_path[0]
 
 def create_demo(model_name: str, device: str = "cuda" if torch.cuda.is_available() else "cpu", offload: bool = False):
     editor = FluxEditor(args)
@@ -505,10 +505,11 @@ def create_demo(model_name: str, device: str = "cuda" if torch.cuda.is_available
                     inject_step = gr.Slider(0, 15, 1, step=1, label="Number of inject steps")
                 output_image = gr.Image(label="Generated/Edited Image")
                 reset_btn = gr.Button("Reset")
+                example_image = gr.Image(label="example Image", visible=False, width=200)
 
         gallery.select(on_select, gallery, [init_image, source_prompt])
         gallery.upload(on_upload, gallery, init_image)
-        init_image.change(on_change, init_image, gallery)
+        example_image.change(on_change, example_image, [gallery, init_image])
 
         generate_btn.click(
             fn=editor.process_image,
@@ -521,7 +522,7 @@ def create_demo(model_name: str, device: str = "cuda" if torch.cuda.is_available
         gr.Examples(
             examples=examples,
             inputs=[
-                init_image, 
+                example_image, 
                 source_prompt, 
                 target_prompt, 
                 editing_strategy,