#!/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()