Update app.py

app.py

@@ -3,6 +3,7 @@ import warnings
 from pathlib import Path
 
 import gradio as gr
+from gradio_imageslider import ImageSlider  # Import ImageSlider
 import torch
 import fastai
 from deoldify import device
@@ -35,7 +36,7 @@ def colorizer_fn(input_img):
             watermarked=False,
             post_process=True,
         )
-        return output_img
+        return (input_img, output_img)  # Return tuple for ImageSlider
     else:
         return None
 
@@ -48,15 +49,15 @@ footer {display: none !important;} /* Hide footer */
 button[title="Share"] { display: none !important; } /* Hides the Share button */
 """
 
-# Gradio Interface (Button to select image, shows only output)
+# Gradio Interface (Image uploader + Slider output)
 with gr.Blocks(css=custom_css) as demo:
-    gr.Markdown("# AI Image Colorizer")
+    gr.Markdown("# AI Image Colorizer with Slider")
 
     # File uploader
     image_input = gr.File(label="Select Image", file_types=[".png", ".jpg", ".jpeg"])
     
-    # Fixed-size output display
-    output = gr.Image(label="Colorized Image", type="filepath", elem_id="fixed-output")
+    # Image Slider for comparison
+    output = ImageSlider(label="Old vs Colorized Image", type="filepath")
 
     # Process image when uploaded
     image_input.change(colorizer_fn, inputs=image_input, outputs=output)