Update app.py

app.py

@@ -187,6 +187,7 @@ def infer(edit_images,
           true_cfg_scale=1.0,
           rewrite_prompt=True,
           progress=gr.Progress(track_tqdm=True)):
+    
     image = edit_images["background"]
     mask = edit_images["layers"][0]
     
@@ -209,7 +210,7 @@ def infer(edit_images,
         generator=torch.Generator(device="cuda").manual_seed(seed)
     ).images[0]
     
-    return result_image, seed
+    return [image,result_image], seed
     
 examples = [
     "change the hat to red",
@@ -277,7 +278,7 @@ with gr.Blocks(css=css) as demo:
                 )
                 run_button = gr.Button("Run")
                 
-            result = gr.Image(label="Result", show_label=False)
+            result = gr.ImageSlider(label="Result", show_label=False)
         
         with gr.Accordion("Advanced Settings", open=False):