Add encoding logic and remove image decoding logic

handler.py

@@ -28,7 +28,6 @@ class EndpointHandler():
             A :obj:`dict`:. base64 encoded image
         """
         inputs = data.pop("inputs", data)
-        encoded_image = data.pop("image", None)
         params = data.pop("parameters", data)
 
         # hyperparamters
@@ -41,12 +40,8 @@ class EndpointHandler():
 
         generator = torch.Generator(device).manual_seed(manual_seed)
 
-        if encoded_image is not None:
-            image = self.decode_base64_image(encoded_image)
-
         # run inference pipeline
         out = self.pipe(inputs, 
-                        image=image,             
                         generator=generator,             
                         num_inference_steps=num_inference_steps,
                         guidance_scale=guidance_scale,
@@ -57,11 +52,7 @@ class EndpointHandler():
         )
             
         # return first generate PIL image
-        return out.images[0]
-    
-    # helper to decode input image
-    def decode_base64_image(self, image_string):
-        base64_image = base64.b64decode(image_string)
-        buffer = BytesIO(base64_image)
-        image = Image.open(buffer)
-        return image
+        image = out.images[0]
+        buffered = BytesIO()
+        image.save(buffered, format="JPEG")
+        return base64.b64encode(buffered.getvalue())