col order

app.py

@@ -89,8 +89,7 @@ def get_folders_matching_format(directory):
     return folders
 
 
-def get_unique_column_names(all_data):
-
+def get_unique_column_names(data=None):
     return [
         "ao3_\u200benglish",
         "bbc_\u200bnews",
@@ -146,13 +145,15 @@ def update_table(
 
     combined_data["Name"] = combined_data["Name"].apply(lambda x: x.replace(".pth", ""))
 
-    relevant_columns = [col for col in visible_columns if col not in ["Name", "Parameters Count (B)", "Average (The lower the better)"]]
+    ordered_columns = get_unique_column_names()
+    relevant_columns = [col for col in ordered_columns if col in visible_columns and col not in ["Name", "Parameters Count (B)", "Average (The lower the better)"]]
+    
     if len(combined_data) > 0:
         combined_data["Average (The lower the better)"] = round(combined_data[relevant_columns].mean(axis=1), 3)
     combined_data = combined_data.rename(columns={"Parameters Count (B)": "Params (B)"})
     combined_data = combined_data.rename(columns={"Average (The lower the better)": "Average (lower=better)"})
     sorted_data = combined_data.sort_values(by=sort_by, ascending=ascending)
-    visible_columns = ["Name", "Params (B)", "Average (lower=better)"] + visible_columns
+    visible_columns = ["Name", "Params (B)", "Average (lower=better)"] + relevant_columns
     filtered_data = sorted_data[visible_columns]
     filtered_data.columns = [col.replace("_", " ") for col in filtered_data.columns]
 
@@ -578,7 +579,7 @@ with gr.Blocks(css=css) as demo:
                 with gr.Column():
                     period_selector = gr.Dropdown(label="Period", choices=time_list, value=last_period)
                     model_selector = gr.CheckboxGroup(label="Model Size", choices=model_size_list, value=model_size_list)
-                    size_range_slider = RangeSlider(minimum=0, maximum=15, value=[0, 15], step=0.1, label="Model Size Range")
+                    size_range_slider = RangeSlider(minimum=0, maximum=40, value=[0, 40], step=0.1, label="Model Size Range")
                     metric_selector = gr.Dropdown(label="Metric", choices=metric_list, value=initial_metric)
                 with gr.Column():
                     midpoint_slider = gr.Slider(minimum=0.1, maximum=0.9, value=0.5, step=0.01, label="Color Gradient Midpoint")