add logger

main.py

@@ -3,7 +3,7 @@ from fastapi import FastAPI,Body
 
 import uvicorn
 import json
-
+import logging
 from PIL import Image
 import time
 from constants import DESCRIPTION, LOGO
@@ -56,7 +56,9 @@ async def predict(prompt=Body(...),imgbase64data=Body(...),userId=Body(None)):
     print("s生成完成:", end2 - end1)    
     # 将图片对象转换为bytes
     output_image_base64 = base64.b64encode(output_image.tobytes()).decode()
-    print("完成的图片:", output_image_base64) 
+    print("完成的图片:", output_image_base64)
+    logger = logging.getLogger('')
+    logger.info(output_image_base64)
     return output_image_base64