correct

app.py

@@ -35,19 +35,26 @@ def load_submissions():
     challenges = defaultdict(list)
     categories = defaultdict(list)
 
+    total_submissions = 0
+
     for _id in relevant_ids:
         ds = load_dataset(_id)["train"]
         for result, image_id in zip(ds["result"], ds["id"]):
+            if result not in submission_names:
+                # Make sure that incorrect model names are not added
+                continue
+
             ids[result].append(image_id)
             challenges[parti_prompt_challenge[image_id]].append(result)
             categories[parti_prompt_categories[image_id]].append(result)
+            total_submissions += 1
     
     all_values = sum(len(v) for v in ids.values())
     main_dict = {k: '{:.2%}'.format(len(v)/all_values) for k, v in ids.items()}
     challenges = {k: Counter(v) for k, v in challenges.items()}
     categories = {k: Counter(v) for k, v in categories.items()}
 
-    return main_dict, challenges, categories
+    return total_submissions, main_dict, challenges, categories
 
 def sort_by_highest_percentage(df):
     # Convert percentage values to numeric format
@@ -56,7 +63,7 @@ def sort_by_highest_percentage(df):
     return df
 
 def get_dataframe_all():
-    main, challenges, categories = load_submissions()
+    total_submissions, main, challenges, categories = load_submissions()
     main_frame = pd.DataFrame([main])
 
     challenges_frame = pd.DataFrame.from_dict(challenges).fillna(0).T
@@ -75,7 +82,7 @@ def get_dataframe_all():
     categories_frame = categories_frame.reset_index().rename(columns={'index': 'Category'})
     challenges_frame = challenges_frame.reset_index().rename(columns={'index': 'Challenge'})
 
-    return main_frame, challenges_frame, categories_frame
+    return total_submissions, main_frame, challenges_frame, categories_frame
 
 TITLE = "# Open Parti Prompts Leaderboard"
 DESCRIPTION = """
@@ -113,7 +120,7 @@ with gr.Blocks() as demo:
     headers = list(SUBMISSIONS.keys())
     datatype = "str"
 
-    main_df, challenge_df, category_df = get_dataframe_all()
+    total_submissions, main_df, challenge_df, category_df = get_dataframe_all()
 
     with gr.Column():
         gr.Markdown("# Open Parti Prompts")
@@ -148,8 +155,12 @@ with gr.Blocks() as demo:
             interactive=False,
         )
 
+    with gr.Column():
+        gr.Markdown("## # Submissions")
+        num_submissions = gr.Number(value=total_submissions, interactive=False)
+
     with gr.Row():
         refresh_button = gr.Button("Refresh")
-        refresh_button.click(refresh, inputs=[], outputs=[main_dataframe, cat_dataframe, chal_dataframe])
+        refresh_button.click(refresh, inputs=[], outputs=[num_submissions, main_dataframe, cat_dataframe, chal_dataframe])
 
 demo.launch()