Update app.py

app.py

@@ -23,8 +23,10 @@ MODEL_KEYS = "-".join(SUBMISSIONS.keys())
 SUBMISSION_ORG = f"results-{MODEL_KEYS}"
 
 submission_names = list(SUBMISSIONS.keys())
-parti_prompt_categories = load_dataset(os.path.join(ORG, "sd-v1-5"))["train"]["Category"]
-parti_prompt_challenge = load_dataset(os.path.join(ORG, "sd-v1-5"))["train"]["Challenge"]
+ds = load_dataset("nateraw/parti-prompts")["train"]
+
+parti_prompt_categories = ds["Category"]
+parti_prompt_challenge = ds["Challenge"]
 
 
 def load_submissions():