Update handler.py

handler.py

@@ -96,7 +96,7 @@ class EndpointHandler():
         self.pipe.enable_xformers_memory_efficient_attention()
         
         # run inference pipeline
-        out = self.pipe(prompt=prompt, negative_prompt=negative_prompt, image=image, mask_image=mask_image)
+        out = self.pipe(prompt=prompt, negative_prompt=negative_prompt, image=image, mask_image=mask_image, height=image.heigth, width=image.width)
 
         print("1st pipeline part successful!")