INNER CODE UNIT · Python

tagger

zhulinyv/Semi-Auto-NovelAI-to-Pixiv · src/batch_tagger.py:44

def tagger(
    image,
    path,
    batch,
    model_repo,
    general_thresh,
    general_mcut_enabled,
    character_thresh,
    character_mcut_enabled,
):
    client = Client("SmilingWolf/wd-tagger", verbose=False)

    if batch:
        imgs_list = file_namel2pathl(file_path2list(path), Path(path))
    else:
        image.save("./output/temp.png")
        imgs_list = ["./output/temp.png"]
    for img in imgs_list:

View source record →