patrickvonplaten commited on
Commit
8f03ebf
1 Parent(s): 2ec8da5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -23,8 +23,10 @@ MODEL_KEYS = "-".join(SUBMISSIONS.keys())
23
  SUBMISSION_ORG = f"results-{MODEL_KEYS}"
24
 
25
  submission_names = list(SUBMISSIONS.keys())
26
- parti_prompt_categories = load_dataset(os.path.join(ORG, "sd-v1-5"))["train"]["Category"]
27
- parti_prompt_challenge = load_dataset(os.path.join(ORG, "sd-v1-5"))["train"]["Challenge"]
 
 
28
 
29
 
30
  def load_submissions():
23
  SUBMISSION_ORG = f"results-{MODEL_KEYS}"
24
 
25
  submission_names = list(SUBMISSIONS.keys())
26
+ ds = load_dataset("nateraw/parti-prompts")["train"]
27
+
28
+ parti_prompt_categories = ds["Category"]
29
+ parti_prompt_challenge = ds["Challenge"]
30
 
31
 
32
  def load_submissions():