Update app.py

app.py

@@ -36,14 +36,14 @@ def filter_data(model, scenario):
     return pd.DataFrame([pass_at_k])
 
 # Initialize the leaderboard
-def init_leaderboard(dataframe, height=600):
+def init_leaderboard(dataframe, default_selection=["Model", "pass@1", "pass@5", "pass@10"], height=600):
     if dataframe is None or dataframe.empty:
         raise ValueError("Leaderboard DataFrame is empty or None.")
     return Leaderboard(
         value=dataframe,
         datatype=["markdown", "number", "number", "number"],  # Specify the types of your columns
         select_columns=SelectColumns(
-            default_selection=["Model", "pass@1", "pass@5", "pass@10"],  # Columns to display by default
+            default_selection=default_selection,  # Columns to display by default
             cant_deselect=[],  # Columns that cannot be deselected
             label="Select Columns to Display:",
         ),
@@ -81,7 +81,7 @@ with demo:
         'pass@10': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 10).mean()
     }, index=['pass@1', 'pass@5', 'pass@10'])).reset_index()
 
-    duo_leaderboard = init_leaderboard(duo_complete_pass_at_k, height=800)    
+    duo_leaderboard = init_leaderboard(duo_complete_pass_at_k, default_selection = ["Model", "pass@1"], height=400)    
     leaderboard = init_leaderboard(complete_pass_at_k, height=800)
 
 # Launch the Gradio interface