Make display_results robust

app.py

@@ -130,8 +130,11 @@ def load_results_dataframes(*model_ids):
     return [load_results_dataframe(model_id) for model_id in model_ids]
 
 
-def display_results(df_1, df_2, task):
-    df = pd.concat([df.set_index("index") for df in [df_1, df_2] if "index" in df.columns])
+def display_results(task, *dfs):
+    dfs = [df.set_index("index") for df in dfs if "index" in df.columns]
+    if not dfs:
+        return None, None
+    df = pd.concat(dfs)
     df = df.T.rename_axis(columns=None)
     return display_tab("results", df, task), display_tab("configs", df, task)
 
@@ -299,7 +302,7 @@ with gr.Blocks(fill_height=True) as demo:
         outputs=[dataframe_1, dataframe_2],
     ).then(
         fn=display_results,
-        inputs=[dataframe_1, dataframe_2, task],
+        inputs=[task, dataframe_1, dataframe_2],
         outputs=[results, configs],
     ).then(
         fn=update_tasks_component,
@@ -307,7 +310,7 @@ with gr.Blocks(fill_height=True) as demo:
     )
     task.change(
         fn=display_results,
-        inputs=[dataframe_1, dataframe_2, task],
+        inputs=[task, dataframe_1, dataframe_2],
         outputs=[results, configs],
     )