new image slider, output as png

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import spaces
-from gradio_imageslider import ImageSlider
 from image_gen_aux import UpscaleWithModel
 from image_gen_aux.utils import load_image
 from fastapi.middleware.cors import CORSMiddleware
@@ -16,7 +15,7 @@ MODELS = {
 @spaces.GPU
 def upscale_image(image, model_selection):
     original = load_image(image)
-
+    
     upscaler = UpscaleWithModel.from_pretrained(MODELS[model_selection]).to("cuda")
     image = upscaler(original, tiling=True, tile_width=1024, tile_height=1024)
 
@@ -45,9 +44,10 @@ with gr.Blocks() as demo:
 
             run_button = gr.Button("Upscale")
         with gr.Column():
-            result = ImageSlider(
+            result = gr.ImageSlider(
                 interactive=False,
                 label="Generated Image",
+                format="png"
             )
 
     run_button.click(