Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from gradio_client import Client, handle_file
 
+from gradio_imageslider import ImageSlider
+
 def get_flux_image(prompt):
     client = Client("black-forest-labs/FLUX.1-schnell")
     result = client.predict(
@@ -39,7 +41,7 @@ def get_upscale(prompt, img_path):
 def main(prompt):
     step_one_flux = get_flux_image(prompt)
     step_two_upscale = get_upscale(prompt, step_one_flux)
-    return step_two_upscale
+    return (step_one_flux, step_two_upscale)
 
 css = """
 #col-container{
@@ -52,7 +54,7 @@ with gr.Blocks(css=css) as demo:
         gr.Markdown("Flux Upscaled")
         prompt_in = gr.Textbox(label="Prompt")
         submit_btn = gr.Button("Submit")
-        output_res = gr.Image(label="Result")
+        output_res = ImageSlider(label="Flux / Upscaled")
 
     submit_btn.click(
         fn=main,