OpenGenAI/open-parti-prompts

1#!/usr/bin/env python32from datasets import load_dataset3from collections import Counter4from random import choices, shuffle5from pandas import DataFrame6import os7import gradio as gr8 9parti_prompt_results = []10ORG = "diffusers-parti-prompts"11SUBMISSIONS = {12    "sd_v1_5": load_dataset(os.path.join(ORG, "sd-v1-5"))["train"],13    "sd_v2_1": load_dataset(os.path.join(ORG, "sd-v2.1"))["train"],14    "if_v1_0": load_dataset(os.path.join(ORG, "karlo-v1"))["train"],15    "karlo": load_dataset(os.path.join(ORG, "if-v-1.0"))["train"],16    # "Kadinsky":17}18import ipdb; ipdb.set_trace()19