update handler file

handler.py

@@ -11,6 +11,7 @@ import cv2
 
 
 import torch
+import base64
 torch.cuda.empty_cache()
 torch.cuda.set_per_process_memory_fraction(0.5)
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:32,garbage_collection_threshold:0.7"
@@ -43,5 +44,7 @@ class EndpointHandler:
 
         img_byte_arr = BytesIO()
         output = Image.fromarray(output)
+        print(output.size)
         output.save(img_byte_arr, format='PNG')
-        return img_byte_arr.getvalue()
+        img_str = base64.b64encode(img_byte_arr.getvalue())
+        return {"out_image": img_str}