sasha HF staff commited on
Commit
d087156
β€’
1 Parent(s): 7ddc374
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -34,6 +34,7 @@ def get_model_names(task_data):
34
  all_df=all_df.drop_duplicates(subset=['model'])
35
  all_df['parameters'] = all_df['parameters'].apply(format_params)
36
  all_df['model'] = all_df['model'].apply(make_link)
 
37
  all_df['Total GPU Energy (Wh)'] = all_df['Total GPU Energy (Wh)'].round(2)
38
  model_names = all_df[['model','parameters','Total GPU Energy (Wh)']]
39
  print(model_names)
 
34
  all_df=all_df.drop_duplicates(subset=['model'])
35
  all_df['parameters'] = all_df['parameters'].apply(format_params)
36
  all_df['model'] = all_df['model'].apply(make_link)
37
+ all_df['Total GPU Energy (Wh)'] = all_df['total_gpu_energy']*1000
38
  all_df['Total GPU Energy (Wh)'] = all_df['Total GPU Energy (Wh)'].round(2)
39
  model_names = all_df[['model','parameters','Total GPU Energy (Wh)']]
40
  print(model_names)