Update app.py

app.py

@@ -13,10 +13,10 @@ import secrets
 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"],
+    "sd-v1-5": load_dataset(os.path.join(ORG, "sd-v1-5"), keep_in_memory=True)["train"],
+    "sd-v2-1": load_dataset(os.path.join(ORG, "sd-v2.1"), keep_in_memory=True)["train"],
+    "if-v1-0": load_dataset(os.path.join(ORG, "karlo-v1"), keep_in_memory=True)["train"],
+    "karlo": load_dataset(os.path.join(ORG, "if-v-1.0"), keep_in_memory=True)["train"],
     # "Kadinsky":
 }