Update app.py

app.py

@@ -392,7 +392,7 @@ with block:
                 selected_columns = [item for item in TASK_V2_INFO if item in selected_columns]
                 present_columns = MODEL_INFO_V2 + selected_columns
                 updated_data = updated_data[present_columns]
-                updated_data = updated_data.sort_values(by="Avg. P1", ascending=False)
+                updated_data = updated_data.sort_values(by=selected_columns[0], ascending=False)
                 updated_headers = present_columns
                 update_datatype = [DATA_TITILE_V2_TYPE[COLUMN_V2_NAMES.index(x)] for x in updated_headers]
 
@@ -489,7 +489,7 @@ with block:
                 selected_columns = [item for item in TASK_INFO if item in selected_columns]
                 present_columns = MODEL_INFO + selected_columns
                 updated_data = updated_data[present_columns]
-                updated_data = updated_data.sort_values(by="Avg. All", ascending=False)
+                updated_data = updated_data.sort_values(by=selected_columns[0], ascending=False)
                 updated_headers = present_columns
                 update_datatype = [DATA_TITILE_TYPE[COLUMN_NAMES.index(x)] for x in updated_headers]