changing color

app.py

@@ -18,7 +18,7 @@ def get_plots(task):
     all_df = all_df.sort_values(by=['Total GPU Energy (Wh)'])
     all_df['parameters'] = all_df['parameters'].apply(format_params)
     all_df['energy_star'] = pd.cut(all_df['Total GPU Energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])
-    fig = px.scatter(all_df, x="model", y='Total GPU Energy (Wh)', custom_data=['parameters'], height= 500, width= 800, color = 'energy_star', color_discrete_map={"⭐": 'red', "⭐⭐": "yellow", "⭐⭐⭐": "green"})
+    fig = px.scatter(all_df, x="model", y='Total GPU Energy (Wh)', custom_data=['parameters'], height= 500, width= 800, color = 'energy_star', color_discrete_map={"⭐": 'red', "⭐⭐": "orange", "⭐⭐⭐": "green"})
     fig.update_traces(
     hovertemplate="<br>".join([
         "Total Energy: %{y}",
@@ -39,7 +39,7 @@ def get_all_plots():
         all_df = pd.concat([all_df, tasks_df])
     all_df = all_df.sort_values(by=['Total GPU Energy (Wh)'])
     all_df['parameters'] = all_df['parameters'].apply(format_params)
-    fig = px.scatter(all_df, x="model", y='Total GPU Energy (Wh)', custom_data=['parameters'], height= 500, width= 800, color = 'energy_star', color_discrete_map={"⭐": 'red', "⭐⭐": "yellow", "⭐⭐⭐": "green"})
+    fig = px.scatter(all_df, x="model", y='Total GPU Energy (Wh)', custom_data=['parameters'], height= 500, width= 800, color = 'energy_star', color_discrete_map={"⭐": 'red', "⭐⭐": "orange", "⭐⭐⭐": "green"})
     fig.update_traces(
     hovertemplate="<br>".join([
         "Total Energy: %{y}",