Update handler.py

handler.py

@@ -55,7 +55,7 @@ class EndpointHandler():
         # run inference pipeline
         out = self.pipe(prompt=prompt, image=image, mask_image=mask_image)
 
-        #image = out.images[0].resize((1024, 1024))
+        image = out.images[0].resize((1024, 1024))
 
         self.pipe2.enable_xformers_memory_efficient_attention()