Update app.py

app.py

@@ -433,7 +433,7 @@ def inference(prompt, ref_images, seed, size_level):
     )[0]
     
     print(f"Time taken: {time.time() - start_time:.2f} seconds")
-    return image, random_seed
+    return (ref_images, image), random_seed
 
 with gr.Blocks() as demo:
     gr.Markdown(
@@ -456,7 +456,8 @@ with gr.Blocks() as demo:
             generate_btn = gr.Button("Generate")
 
         with gr.Column():
-            output_image = gr.Image(label="Generated Image",type='pil',image_mode='RGB')
+            #output_image = gr.Image(label="Generated Image",type='pil',image_mode='RGB')
+            output_image = gr.ImageSlider(label="Generated Image", type="pil", image_mode='RGB')
             output_random_seed = gr.Textbox(label="Used Seed", lines=5)
     from functools import partial
     generate_btn.click(