Update handler.py

handler.py

@@ -22,9 +22,10 @@ class EndpointHandler():
         
         images = self.pipe(prompt, image=raw_images, num_inference_steps=25, image_guidance_scale=1).images
         img=images[0]
-        img_data=img.tobytes()
-        img_base64=base64.b64encode(img_data).decode('utf-8')
-        return {'image':img_base64}
+        img.save("./1.png")
+        with open('./1.png','rb') as img_file:
+            encoded_string = base64.b64encode(img_file.read()).decode('utf-8')
+        return {'image':encoded_string}