|
|
|
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"], |
|
|
|
} |
|
import ipdb; ipdb.set_trace() |
|
|
