Update handler for testing

handler.py

@@ -2,8 +2,6 @@ from typing import  Dict, List, Any
 import torch
 from diffusers import StableDiffusionPipeline, EulerAncestralDiscreteScheduler
 from PIL import Image
-import base64
-from io import BytesIO
 
 
 # set device
@@ -52,7 +50,4 @@ class EndpointHandler():
         )
             
         # return first generate PIL image
-        image = out.images[0]
-        buffered = BytesIO()
-        image.save(buffered, format="JPEG")
-        return base64.b64encode(buffered.getvalue())
+        return out.images[0]