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"), 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"],
+    "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":
 }
 
@@ -76,7 +76,7 @@ def load_submissions():
     relevant_ids = [d.id for d in all_datasets]
     
     submitted_ids = []
-    for _id in relevant_ids:
+    for _id in relevant_ids[:2]:
         ds = load_dataset(_id)["train"]
         submitted_ids += ds["id"]