Function deoldify_api

scripts/api.py:24–39

(_: gr.Blocks, app: FastAPI)

warnings.filterwarnings("ignore", category=FutureWarning, message="Arguments other than a weight enum or `None`.*?")

def deoldify_api(_: gr.Blocks, app: FastAPI):
    @app.post("/deoldify/image")
    async def deoldify_image(
        input_image: str = Body("",title="image input"),
        render_factor: int = Body(35,title="render factor"),
        artistic: bool = Body(False,title="artistic")
    ):
        vis = get_image_colorizer(root_folder=Path(paths_internal.models_path),render_factor=render_factor, artistic=artistic)
        # 判断input_image是否是url
        if input_image.startswith("http"):
            img = vis._get_image_from_url(input_image)
        else:
            # 把base64转换成图片 PIL.Image
            img = Image.open(BytesIO(base64.b64decode(input_image)))
        outImg = vis.get_transformed_image_from_image(img, render_factor=render_factor)
        return {"image": api.encode_pil_to_base64(outImg).decode("utf-8")}

try:
    import modules.script_callbacks as script_callbacks