Update app.py

app.py

@@ -428,12 +428,14 @@ with gr.Blocks(css=css, fill_height=True) as demo:
             
                 with gr.Row(equal_height=False):
                     input_image = gr.ImageMask(
-                        type="pil", label="Input Image", crop_size=(1024, 1024), layers=False
+                        type="pil", label="Input Image", layers=False
+                        # type="pil", label="Input Image", crop_size=(1024, 1024), layers=False
                     )
             
                     result = ImageSlider(
                         interactive=False,
                         label="Generated Image",
+                        format="png"
                     )
             
                 use_as_input_button = gr.Button("Use as Input Image", visible=False)
@@ -579,6 +581,7 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                         result = ImageSlider(
                             interactive=False,
                             label="Generated Image",
+                            format="png"
                         )
                         use_as_input_button = gr.Button("Use as Input Image", visible=False)