Update app.py

app.py

@@ -43,18 +43,12 @@ def get_leaderboard_for_task(task_pretty: str) -> gr.components.Dataframe:
         datatype=get_types_per_task(TASKS_PRETTY_REVERSE[task_pretty]),
     )
 
-code_completion_dataset_names = (
-    get_results_for_task(TASKS_PRETTY['project_code_completion'])['Dataset'].unique().tolist()
-)
-cc_ds_names_html = {gr.HTML(ds_name): ds_name for ds_name in code_completion_dataset_names}
+code_completion_dataset_names = get_results_for_task(TASKS_PRETTY['project_code_completion'])['Dataset_Name'].unique().tolist()
 
-def get_leaderboard_for_completion_task(dataset_name: str | None | gr.HTML):
+def get_leaderboard_for_completion_task(dataset_name: str | None):
     df = get_results_for_task(TASKS_PRETTY['project_code_completion'])
-    code_completion_dataset_names = df['Dataset'].unique()
     if dataset_name is None:
         dataset_name = code_completion_dataset_names[0]
-    if isinstance(dataset_name, gr.HTML):
-        dataset_name = cc_ds_names_html[dataset_name]
     filtered_df = df[df['Dataset']==dataset_name]
     return gr.components.Dataframe(
         value=filtered_df,
@@ -86,7 +80,7 @@ with gr.Blocks() as demo:
                 task_input = gr.Text(value=task_pretty, visible=False)
 
                 if task_pretty == TASKS_PRETTY['project_code_completion']:
-                    dataset_dropdown = gr.Dropdown(choices=list(cc_ds_names_html), label="Select the Dataset")
+                    dataset_dropdown = gr.Dropdown(choices=code_completion_dataset_names, label="Select the Dataset")
                     dataset_dropdown.change(
                         fn=get_leaderboard_for_completion_task,
                         inputs=dataset_dropdown,