Update app.py

app.py

@@ -1,8 +1,51 @@
 from typing import List
+import io
+from PIL import Image
 from pydantic import BaseModel
 from lama_cleaner.server import process
-import uvicorn
-from fastapi import FastAPI
+from lama_cleaner.server import main#先初始化才能用process
+#urllib3 1.26.4  兼容
+class FakeArgs(BaseModel):
+    host: str = "0.0.0.0"
+    port: int = 7860
+    model: str = 'lama'
+    hf_access_token: str = ""
+    sd_enable_xformers: bool = False
+    sd_disable_nsfw: bool = False
+    sd_cpu_textencoder: bool = True
+    sd_controlnet: bool = False
+    sd_controlnet_method: str = "control_v11p_sd15_canny"
+    sd_local_model_path: str = ""
+    sd_run_local: bool = False
+    local_files_only: bool = False
+    cpu_offload: bool = False
+    device: str = "cpu"
+    gui: bool = False
+    gui_size: List[int] = [1000, 1000]
+    input: str = ''
+    disable_model_switch: bool = True
+    debug: bool = False
+    no_half: bool = False
+    disable_nsfw: bool = False
+    enable_xformers: bool = False
+    enable_interactive_seg: bool = True
+    interactive_seg_model: str = "vit_b"
+    interactive_seg_device: str = "cpu"
+    enable_remove_bg: bool = False
+    enable_anime_seg: bool = False
+    enable_realesrgan: bool = False
+    enable_gfpgan: bool = False
+    gfpgan_device: str = "cpu"
+    enable_restoreformer: bool = False
+    enable_gif: bool = False
+    quality: int = 95
+    model_dir: str = None
+    output_dir: str = None
+
+#初始化model,已经排除了flask部分
+main(FakeArgs())
+
+
 app = FastAPI()
 @app.on_event("startup")
 async def app_start():
@@ -52,6 +95,12 @@ async def app_start():
         }#payload用data
     
     resp = process(files=files,payload=payload)
-    print(resp)
+    
+    # 从 _io.BytesIO 对象中读取字节数据
+    resp_bytes = resp.read()
+    # 使用 Image.open() 方法打开图片
+    img = Image.open(io.BytesIO(resp_bytes))
+    img.show()
+
 if __name__ == '__main__':
     uvicorn.run(app, host='0.0.0.0', port=7860)