pufanyi commited on
Commit
b7f775a
·
1 Parent(s): e7b07cd

Refactor GET_DEFAULT_CHECKBOX function to handle exceptions in app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -11
app.py CHANGED
@@ -116,17 +116,20 @@ def init_leaderboard(dataframes):
116
  print("Subset:", subset)
117
  print("Search Term:", search_term)
118
  print("Selected Columns:", selected_columns)
119
- filtered_data = dataframes[subset]
120
- if search_term:
121
- filtered_data = filtered_data[dataframes[subset]["Model Name"].str.contains(search_term, case=False)]
122
- filtered_data.sort_values(by="Total", ascending=False, inplace=True)
123
- # selected_columns = [c.name for c in fields(AutoEvalColumn) if c.name in selected_columns]
124
- selected_data = filtered_data[selected_columns]
125
- return gr.DataFrame(
126
- selected_data,
127
- type="pandas",
128
- # datatype=[c.type for c in fields(AutoEvalColumn) if c.name in selected_columns],
129
- )
 
 
 
130
 
131
  with gr.Row():
132
  refresh_button = gr.Button("Refresh")
 
116
  print("Subset:", subset)
117
  print("Search Term:", search_term)
118
  print("Selected Columns:", selected_columns)
119
+ try:
120
+ filtered_data = dataframes[subset]
121
+ if search_term:
122
+ filtered_data = filtered_data[dataframes[subset]["Model Name"].str.contains(search_term, case=False)]
123
+ filtered_data.sort_values(by="Total", ascending=False, inplace=True)
124
+ # selected_columns = [c.name for c in fields(AutoEvalColumn) if c.name in selected_columns]
125
+ selected_data = filtered_data[selected_columns]
126
+ return gr.DataFrame(
127
+ selected_data,
128
+ type="pandas",
129
+ # datatype=[c.type for c in fields(AutoEvalColumn) if c.name in selected_columns],
130
+ )
131
+ except Exception as e:
132
+ return refresh(subset)
133
 
134
  with gr.Row():
135
  refresh_button = gr.Button("Refresh")