Update handler.py

handler.py

@@ -51,10 +51,7 @@ class EndpointHandler:
             controlnet_conditioning_scale=controlnet_conditioning_scale,
         ).images[0]
 
-        # Save the output image to bytes
-        output_bytes = save_image_to_bytes(output_image)
-
-        return output_bytes
+        return output_image
 
     def make_inpaint_condition(self, image, mask):
         image = np.array(image.convert("RGB")).astype(np.float32) / 255.0