open-parti-prompts / verify.py
patrickvonplaten's picture
up
88fd33d
raw
history blame
581 Bytes
#!/usr/bin/env python3
from datasets import load_dataset
from collections import Counter
from random import choices, shuffle
from pandas import DataFrame
import os
import gradio as gr
parti_prompt_results = []
ORG = "diffusers-parti-prompts"
SUBMISSIONS = {
"sd_v1_5": load_dataset(os.path.join(ORG, "sd-v1-5"))["train"],
"sd_v2_1": load_dataset(os.path.join(ORG, "sd-v2.1"))["train"],
"if_v1_0": load_dataset(os.path.join(ORG, "karlo-v1"))["train"],
"karlo": load_dataset(os.path.join(ORG, "if-v-1.0"))["train"],
# "Kadinsky":
}
import ipdb; ipdb.set_trace()