fix base64

main.py

@@ -55,7 +55,14 @@ async def predict(prompt=Body(...),imgbase64data=Body(...),userId=Body(None)):
     print("测试",output_image)
     print("s生成完成:", end2 - end1)    
     # 将图片对象转换为bytes
-    output_image_base64 = base64.b64encode(output_image.tobytes()).decode()
+    image_data = io.BytesIO()
+
+    # 将图像保存到BytesIO对象中，格式为JPEG
+    output_image.save(image_data, format='JPEG')
+    
+    # 将BytesIO对象的内容转换为字节串
+    image_data_bytes = image_data.getvalue()
+    output_image_base64 = base64.b64encode(image_data_bytes).decode('utf-8')
     print("完成的图片:", output_image_base64)
     logger = logging.getLogger('')
     logger.info(output_image_base64)