Update app.py

app.py

@@ -36,7 +36,6 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, s
     mask_image = dict["mask"]
 
     width, height = init_image.size
-    #mask_image = mask_image.resize(init_image.size, resample=Image.LANCZOS)
     mask_image = mask_image.convert("RGBA")
     data = mask_image.getdata()
     new_data = []
@@ -47,7 +46,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].resize(init_image.size, resample=Image.LANCZOS), gr.update(visible=True)