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

new styling

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -234,6 +234,10 @@ def init_leaderboard(dataframe, visible_columns=None):
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
@@ -735,7 +739,7 @@ with demo:
735
  )
736
  model_type_filter = gr.Dropdown(
737
  choices=[
738
- t.to_str("-") for t in ModelType if t != ModelType.Unknown
739
  ],
740
  label="Access Type",
741
  multiselect=True,
 
234
  dataframe = dataframe.copy()
235
  dataframe["model_type"] = dataframe["model_type"].str.replace(" : ", "-")
236
 
237
+ if "guard_model_type" in dataframe.columns:
238
+ dataframe = dataframe.copy()
239
+ dataframe["guard_model_type"] = dataframe["guard_model_type"].str.replace("wc_guard", "whitecircle_guard")
240
+
241
  # print("\n\n", "dataframe", dataframe, "--------------------------------\n\n")
242
 
243
  # Determine which columns to display
 
739
  )
740
  model_type_filter = gr.Dropdown(
741
  choices=[
742
+ t.to_str("-") for t in ModelType if t != ModelType.Unknown and t != ModelType.ClosedSource
743
  ],
744
  label="Access Type",
745
  multiselect=True,