Update app.py

app.py

@@ -34,6 +34,9 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, s
     
     init_image = dict["image"]
     mask_image = dict["mask"]
+
+    width, height = init_image.size
+    
     mask_image = mask_image.convert("RGBA")
     data = mask_image.getdata()
     new_data = []
@@ -44,7 +47,7 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, s
             new_data.append(item)
     
     mask_image.putdata(new_data)
-    
+    mask_image = mask_image.resize(init_image.size, resample=Image.LANCZOS)
     output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask_image, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength, clip_skip=1)
     return output.images[0], gr.update(visible=True)