sasha HF staff commited on
Commit
a9edcf0
β€’
1 Parent(s): a04a11c
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ def get_model_names(task):
63
  all_df['Total GPU Energy (Wh)'] = all_df['Total GPU Energy (Wh)'].round(2)
64
  all_df['Rating'] = pd.cut(all_df['Total GPU Energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])
65
  all_df= all_df.sort_values('Total GPU Energy (Wh)')
66
- model_names = all_df[['Model','Rating','Total GPU Energy (Wh)']]
67
  return model_names
68
 
69
  def get_all_model_names():
 
63
  all_df['Total GPU Energy (Wh)'] = all_df['Total GPU Energy (Wh)'].round(2)
64
  all_df['Rating'] = pd.cut(all_df['Total GPU Energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])
65
  all_df= all_df.sort_values('Total GPU Energy (Wh)')
66
+ model_names = all_df[['Model','Rating','Total GPU Energy (Wh)', 'Parameters']]
67
  return model_names
68
 
69
  def get_all_model_names():