update handler file

handler.py

@@ -28,7 +28,7 @@ class EndpointHandler:
             tile=0,
             tile_pad=10,
             # pre_pad=args.pre_pad,
-            # half=not args.fp32,
+            half=True,
             # gpu_id=args.gpu_id
         )
 
@@ -46,6 +46,7 @@ class EndpointHandler:
 
         img_byte_arr = BytesIO()
         output = Image.fromarray(output)
+        print(output.size)
         output.save(img_byte_arr, format='PNG')
         img_str = base64.b64encode(img_byte_arr.getvalue())
         return {"out_image": img_str.decode()}