Update app.py

app.py

@@ -88,7 +88,7 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=5, steps=30, str
     mask[mask>0]=255
     mask = Image.fromarray(mask)
     
-    output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength)
+    output = pipe(prompt = prompt,width=w,height=h, negative_prompt=negative_prompt, image=init_image, mask_image=mask, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength)
     
     return output.images[0] #, gr.update(visible=True)