Spaces:
Running
Running
nevermind
Browse files
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():
|