model_compare/gen_markdown.py

import sys

output_types = sys.argv[1:]

print("## ControlNet Canny Comparison")
print("")

# header
print("|seed|", end="")
for ot in output_types:
    print(f"{ot}|", end="")
print("")

print("|", end="")
for _ in range(len(output_types) + 1):
    print("---|", end="")
print("")


def get_url(path):
    #url = "./canny/"
    url = 'https://huggingface.co/takuma104/controlnet_dev/resolve/main/model_compare/canny/'
    return url + path


for seed in range(32):
    print(f"|{seed}|", end="")
    for ot in output_types:
        url = get_url(path=f"{ot}/output_{seed:02d}.png")
        print(f'<a href="{url}"><img width="128" src="{url}"/></a>|', end="")
    print("")