Update app.py

app.py

@@ -15,8 +15,8 @@ def predict(mask_img):
     prompt = "a green frog, highly detailed, natural lighting"
     image = pipeline(prompt=prompt,
                      #num_inference_steps=35,
-                     image=mask_img["image"], 
-                     mask_image=mask_img["mask"] 
+                     image=mask_img["image"].convert("RGB").resize((512, 512)), 
+                     mask_image=mask_img["mask"].convert("RGB").resize((512, 512)) 
                      #guidance_scale=9
                     ).images[0]