Update app.py

app.py

@@ -295,7 +295,7 @@ def infer(input_image, prompt, illumination_dropdown, direction_dropdown, seed=4
         height=input_image.size[1],
         generator=torch.Generator().manual_seed(seed),
     ).images[0]
-    return image, seed, prompt_with_template
+    return [input_image, image], seed, prompt_with_template
 
 def update_prompt_from_dropdown(illumination_option):
     """Update the prompt textbox based on dropdown selection"""
@@ -372,7 +372,7 @@ with gr.Blocks(css=css) as demo:
                     )       
                     
             with gr.Column():
-                result = gr.Image(label="Result", show_label=False, interactive=False)
+                result = gr.ImageSlider(label="Result", show_label=False, interactive=False)
                 final_prompt = gr.Textbox(label="Processed prompt", info="The structure of prompt to use if you download the LoRA")
         # update prompt when dropdown changes
         illumination_dropdown.change(