File size: 581 Bytes
88fd33d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
#!/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()
|