Update app.py

app.py

@@ -63,7 +63,7 @@ def infer(edit_images, prompt, seed=42, randomize_seed=False, width=1024, height
     mask = edit_images["layers"][0].convert("RGB")
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-    image = pipe(
+    out_image = pipe(
         prompt=prompt,
         inpaint_image=image,
         inpaint_mask=mask,
@@ -75,7 +75,7 @@ def infer(edit_images, prompt, seed=42, randomize_seed=False, width=1024, height
         num_inference_steps=num_inference_steps,
         generator=torch.Generator("cpu").manual_seed(seed)
     ).images[0]
-    return image, seed
+    return (image, out_image), seed
     
 examples = [
     "a tiny astronaut hatching from an egg on the moon",
@@ -115,7 +115,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="Generated Image", type="pil", image_mode='RGB')
         
         with gr.Accordion("Advanced Settings", open=False):