changefile

main.py

@@ -14,7 +14,7 @@ from diffusers.utils import load_image
 import base64
 
 app = FastAPI()
-
+pipeline = get_pipeline()
 
 #Endpoints 
 #Root endpoints
@@ -26,7 +26,7 @@ def root():
 async def predict(url:str,prompt:str):
     MAX_QUEUE_SIZE = 4
     start = time.time()
-    pipeline = get_pipeline()
+   
     
     url = "https://img2.baidu.com/it/u=1845675188,2679793929&fm=253&fmt=auto&app=138&f=JPEG?w=667&h=500"
     prompt = "a nice Comfortable and clean. According to Baidu Education Information, the adjectives for a room include: comfortable, clean, beautiful, spacious, warm, quiet, luxurious, pleasant, exquisite, and warm ,colorful, light room width sofa,8k"
@@ -55,10 +55,12 @@ async def predict(url:str,prompt:str):
     print("s生成完成:", end2 - end1)
     
     output_image.save("./imageclm5.png")
-    with open(output_image, 'rb') as image_file:
-        encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
-        print(encoded_string)
-        return encoded_string
+    # 将图片对象转换为bytes
+    image_bytes = output_image.to_bytes() 
+    
+    # 对bytes进行base64编码    
+    encoded_string = base64.b64encode(image_bytes).decode('utf-8')
+    return encoded_string
         
     
 @app.post("/predict")