Update app.py

app.py

@@ -34,12 +34,12 @@ def generate(image_editor, prompt, neg_prompt, strength, guidance):
 
     final_image = pipeline(prompt=prompt, 
                            image=image, 
-                           mask_image=mask, 
-                           width=image.width, 
-                           height=image.height, 
-                           num_inference_steps=50, 
-                           strength=strength, 
-                           guidance_scale=guidance).images[0]
+                           mask_image=mask,).images[0]
+                           #width=image.width, 
+                           #height=image.height, 
+                           #num_inference_steps=50, 
+                           #strength=strength, 
+                           #guidance_scale=guidance).images[0]
 
     return image_editor, image, mask, final_image
 
@@ -64,9 +64,9 @@ with gr.Blocks() as demo:
         guidance_slider = gr.Slider(1.0, 15.0, value=7.5, label="Guidance")
         
     with gr.Row():
-        out1 = gr.Image()
-        out2 = gr.Image()
-        out3 = gr.Image()
+        out1 = gr.Image(format='png')
+        out2 = gr.Image(format='png')
+        out3 = gr.Image(format='png')
         
     generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt, strength_slider, guidance_slider], outputs=[sketch_pad, out1, out2, out3])