adding energy

app.py

@@ -28,11 +28,14 @@ def make_link(mname):
 def get_model_names(task_data):
     #TODO: add link to results in model card of each model
     task_df= pd.read_csv('data/params/'+task_data)
+    energy_df= pd.read_csv('data/energy/'+task_data)
     task_df= task_df.rename(columns={"Link": "model"})
-    task_df=task_df.drop_duplicates(subset=['model'])
-    task_df['parameters'] = task_df['parameters'].apply(format_params)
-    task_df['model'] = task_df['model'].apply(make_link)
-    model_names = task_df[['model','parameters']]
+    all_df = pd.merge(task_df, energy_df, on='model')
+    all_df=all_df.drop_duplicates(subset=['model'])
+    all_df['parameters'] = all_df['parameters'].apply(format_params)
+    all_df['model'] = all_df['model'].apply(make_link)
+    all_df['Total GPU Energy (Wh)'] = all_df['Total GPU Energy (Wh)'].round(2)
+    model_names = all_df[['model','parameters','Total GPU Energy (Wh)']]
     print(model_names)
     return model_names