Update handler.py

handler.py

@@ -16,8 +16,11 @@ class EndpointHandler():
 
         inputs=data['inputs']
         text=inputs.pop('text',data)
-        image = self.pipe(text).images[0] 
-        print(type(image))
-        img_data = image.tobytes()
-        img_base64 = base64.b64encode(img_data).decode('utf-8')
-        return {'image':img_base64}
+        img = self.pipe(text).images[0] 
+        # 将图像转换为字节流
+        img_byte_arr = io.BytesIO()
+        img.save(img_byte_arr, format='JPEG')
+        
+        # 将字节流编码为base64格式
+        encoded_string = base64.b64encode(img_byte_arr.getvalue())
+        return {'image':encoded_string}