rwightman HF staff commited on
Commit
54d6416
·
verified ·
1 Parent(s): 16d5f81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -31,7 +31,6 @@ def load_leaderboard():
31
  dataframes = {name: pd.read_csv(url) for name, url in results_csv_files.items()}
32
  bench_dataframes = {name: pd.read_csv(url) for name, url in benchmark_csv_files.items()}
33
  bench_dataframes = {name: df for name, df in bench_dataframes.items() if 'infer_gmacs' in df.columns}
34
- print(bench_dataframes.keys())
35
 
36
  # Clean up dataframes
37
  remove_column_names = ["top1_err", "top5_err", "top1_diff", "top5_diff", "rank_diff", "param_count"]
@@ -207,9 +206,9 @@ def update_leaderboard_and_plot(
207
  fig = create_scatter_plot(combined_df, x_axis, y_axis, model_name, highlight_name, log_x, log_y)
208
  display_df = combined_df.drop(columns=['highlighted'])
209
  display_df = display_df.style.apply(lambda x: ['background-color: #FFA500' if combined_df.loc[x.name, 'highlighted'] else '' for _ in x], axis=1).format(
210
- {
211
- 'infer_batch_size': lambda x: col_formatter(x), # Integer column
212
- },
213
  precision=2,
214
  )
215
 
 
31
  dataframes = {name: pd.read_csv(url) for name, url in results_csv_files.items()}
32
  bench_dataframes = {name: pd.read_csv(url) for name, url in benchmark_csv_files.items()}
33
  bench_dataframes = {name: df for name, df in bench_dataframes.items() if 'infer_gmacs' in df.columns}
 
34
 
35
  # Clean up dataframes
36
  remove_column_names = ["top1_err", "top5_err", "top1_diff", "top5_diff", "rank_diff", "param_count"]
 
206
  fig = create_scatter_plot(combined_df, x_axis, y_axis, model_name, highlight_name, log_x, log_y)
207
  display_df = combined_df.drop(columns=['highlighted'])
208
  display_df = display_df.style.apply(lambda x: ['background-color: #FFA500' if combined_df.loc[x.name, 'highlighted'] else '' for _ in x], axis=1).format(
209
+ #{
210
+ # 'infer_batch_size': lambda x: col_formatter(x), # Integer column
211
+ #},
212
  precision=2,
213
  )
214