Leaderboard_test

Sleeping

sasha HF Staff commited on Sep 20, 2024

Commit

78be425

1 Parent(s): f4a0e9d

adding rating

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,8 @@ def get_model_names(task_data):
     all_df['model'] = all_df['model'].apply(make_link)
     all_df['Total GPU Energy (Wh)'] = all_df['total_gpu_energy']*1000
     all_df['Total GPU Energy (Wh)'] = all_df['Total GPU Energy (Wh)'].round(2)
-    model_names = all_df[['model','parameters','Total GPU Energy (Wh)']]
     return model_names
 def format_params(num):

     all_df['model'] = all_df['model'].apply(make_link)
     all_df['Total GPU Energy (Wh)'] = all_df['total_gpu_energy']*1000
     all_df['Total GPU Energy (Wh)'] = all_df['Total GPU Energy (Wh)'].round(2)
+    all_df['Rating'] = pd.cut(all_df['Total GPU Energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])
+    model_names = all_df[['model','parameters','Rating', 'Total GPU Energy (Wh)']]
     return model_names
 def format_params(num):