sh1gechan commited on
Commit
fc28ca3
·
verified ·
1 Parent(s): 5cd8dec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -9
app.py CHANGED
@@ -149,8 +149,9 @@ def filter_models(
149
 
150
  type_emoji = [t[0] for t in type_query]
151
  filtered_df = filtered_df.loc[df[AutoEvalColumn.model_type_symbol.name].isin(type_emoji)]
152
- print(f"Unique values in precision column: {filtered_df[AutoEvalColumn.precision.name].unique()}")
153
- filtered_df = filtered_df.loc[filtered_df[AutoEvalColumn.precision.name].isin(precision_query + ["None"])]
 
154
  print(f"After precision filter: {filtered_df.shape}")
155
  filtered_df = filtered_df.loc[df[AutoEvalColumn.add_special_tokens.name].isin(add_special_tokens_query)]
156
  print(f"After add_special_tokens filter: {filtered_df.shape}")
@@ -261,15 +262,11 @@ with demo:
261
  # visible=True,
262
  # #column_widths=["2%", "33%"]
263
  # )
264
- initial_data = leaderboard_df[
265
- [c.name for c in fields(AutoEvalColumn) if c.never_hidden]
266
- + shown_columns.value
267
- ].to_dict('records')
268
-
269
  leaderboard_table = gr.components.Dataframe(
270
  value=initial_data,
271
- headers=[c.name for c in fields(AutoEvalColumn) if c.never_hidden] + shown_columns.value,
272
- datatype=TYPES,
273
  elem_id="leaderboard-table",
274
  interactive=False,
275
  visible=True,
 
149
 
150
  type_emoji = [t[0] for t in type_query]
151
  filtered_df = filtered_df.loc[df[AutoEvalColumn.model_type_symbol.name].isin(type_emoji)]
152
+ if 'Unknown' not in precision_query:
153
+ precision_query.append('Unknown')
154
+ filtered_df = filtered_df.loc[filtered_df[AutoEvalColumn.precision.name].isin(precision_query)]
155
  print(f"After precision filter: {filtered_df.shape}")
156
  filtered_df = filtered_df.loc[df[AutoEvalColumn.add_special_tokens.name].isin(add_special_tokens_query)]
157
  print(f"After add_special_tokens filter: {filtered_df.shape}")
 
262
  # visible=True,
263
  # #column_widths=["2%", "33%"]
264
  # )
265
+ initial_data = leaderboard_df.to_dict('records')
 
 
 
 
266
  leaderboard_table = gr.components.Dataframe(
267
  value=initial_data,
268
+ headers=list(leaderboard_df.columns),
269
+ datatype={col: "str" for col in leaderboard_df.columns}, # すべての列を文字列として扱う
270
  elem_id="leaderboard-table",
271
  interactive=False,
272
  visible=True,