Update app.py

app.py

@@ -32,6 +32,16 @@ SUBMISSION_ORG = f"results-{MODEL_KEYS}"
 PROMPT_FORMAT = "Pick the picture that best matches the prompt: **{}**"
 
 
+all_datasets = list_datasets(author=SUBMISSION_ORG)
+relevant_ids = [d.id for d in all_datasets]
+
+submitted_ids = []
+for _id in relevant_ids:
+    ds = load_dataset(_id)["train"]
+    submitted_ids += ds["id"]
+
+submitted_ids = Counter(submitted_ids)
+
 submission_names = list(SUBMISSIONS.keys())
 num_images = len(SUBMISSIONS[submission_names[0]])
 
@@ -59,16 +69,6 @@ def generate_random_hash(length=8):
 def start():
     ids = {id: 0 for id in range(num_images)}
 
-    # submitted_ids = Counter(submissions["ids"])
-    all_datasets = list_datasets(author=SUBMISSION_ORG)
-    relevant_ids = [d.id for d in all_datasets]
-
-    submitted_ids = []
-    for _id in relevant_ids:
-        ds = load_dataset(_id)["train"]
-        submitted_ids += ds["id"]
-
-    submitted_ids = Counter(submitted_ids)
     ids = {**ids, **submitted_ids}
 
     # sort by count