small UI suggestions

cool demo!
I made some small changes to improve the UI -
- adds Image slider to the output for a cool before & after effect (similar to https://huggingface.co/spaces/RiverZ/ICEdit)
- adjusts the placement of the output

app.py

@@ -81,7 +81,7 @@ def generate_image(lora_name, prompt, uploaded_image, width, height, guidance_sc
     # Clear cache after generation
     clear_cache(pipe.transformer)
 
-    return image
+    return (uploaded_image, image)
 
 # Example data
 examples = [
@@ -117,13 +117,14 @@ def create_gradio_interface():
                 prompt_box = gr.Textbox(label="Prompt", placeholder="Enter a prompt...")
                 image_input = gr.Image(type="pil", label="Upload Image")
             with gr.Column(scale=1):
+                output_image = gr.ImageSlider(label="Generated Image")
                 width_box = gr.Textbox(label="Width", value="1024")
                 height_box = gr.Textbox(label="Height", value="1024")
                 guidance_slider = gr.Slider(minimum=0.1, maximum=20, value=3.5, step=0.1, label="Guidance Scale")
                 steps_slider = gr.Slider(minimum=1, maximum=50, value=25, step=1, label="Inference Steps")
                 seed_slider = gr.Slider(minimum=1, maximum=10000000000, value=42, step=1, label="Seed")
                 generate_button = gr.Button("Generate")
-                output_image = gr.Image(type="pil", label="Generated Image")
+                
         # Add examples for Generation
         gr.Examples(
             examples=examples,