oops

app.py

@@ -7,8 +7,8 @@ def get_plots(task_data):
     task_df= pd.read_csv(task_data)
     task_df['Total GPU Energy (Wh)'] = task_df['total_gpu_energy']*1000
     task_df = task_df.sort_values(by=['Total GPU Energy (Wh)'])
-    task_df['energy_star'] = pd.cut(task_df['total_gpu_energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])
-    task_df = px.scatter(task_df, x="model", y="total_gpu_energy (Wh)", height= 500, width= 800, color = 'energy_star', color_discrete_map={"⭐": 'red', "⭐⭐": "yellow", "⭐⭐⭐": "green"})
+    task_df['energy_star'] = pd.cut(task_df['Total GPU Energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])
+    task_df = px.scatter(task_df, x="model", y='Total GPU Energy (Wh)', height= 500, width= 800, color = 'energy_star', color_discrete_map={"⭐": 'red', "⭐⭐": "yellow", "⭐⭐⭐": "green"})
     return task_df
 
 def get_model_names(task_data):