apsys commited on
Commit
4928594
·
1 Parent(s): e0634f4

new styling

Browse files
Files changed (2) hide show
  1. app.py +5 -0
  2. src/display/utils.py +1 -1
app.py CHANGED
@@ -230,6 +230,10 @@ def init_leaderboard(dataframe, visible_columns=None):
230
  dataframe = dataframe.copy()
231
  dataframe["model_name"] = dataframe["model_name"].str.lower()
232
 
 
 
 
 
233
  # print("\n\n", "dataframe", dataframe, "--------------------------------\n\n")
234
 
235
  # Determine which columns to display
@@ -745,6 +749,7 @@ with demo:
745
  multiselect=True,
746
  value=get_initial_columns(),
747
  interactive=True,
 
748
  scale=1,
749
  )
750
  with gr.Row():
 
230
  dataframe = dataframe.copy()
231
  dataframe["model_name"] = dataframe["model_name"].str.lower()
232
 
233
+ if "model_type" in dataframe.columns:
234
+ dataframe = dataframe.copy()
235
+ dataframe["model_type"] = dataframe["model_type"].str.replace(" : ", "-")
236
+
237
  # print("\n\n", "dataframe", dataframe, "--------------------------------\n\n")
238
 
239
  # Determine which columns to display
 
749
  multiselect=True,
750
  value=get_initial_columns(),
751
  interactive=True,
752
+ visible=False,
753
  scale=1,
754
  )
755
  with gr.Row():
src/display/utils.py CHANGED
@@ -121,7 +121,7 @@ class GuardBenchColumn:
121
  guard_model_type: ColumnInfo = field(default_factory=lambda: ColumnInfo(
122
  name="guard_model_type",
123
  display_name="Type",
124
- displayed_by_default=True
125
  ))
126
  base_model: ColumnInfo = field(default_factory=lambda: ColumnInfo(
127
  name="base_model",
 
121
  guard_model_type: ColumnInfo = field(default_factory=lambda: ColumnInfo(
122
  name="guard_model_type",
123
  display_name="Type",
124
+ displayed_by_default=False
125
  ))
126
  base_model: ColumnInfo = field(default_factory=lambda: ColumnInfo(
127
  name="base_model",