trying to make parameters easier to read

app.py

@@ -11,6 +11,7 @@ def get_plots(task):
     all_df = pd.merge(task_df, params_df, on='model')
     all_df['Total GPU Energy (Wh)'] = all_df['total_gpu_energy']*1000
     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.update_traces(
@@ -27,6 +28,13 @@ def get_model_names(task_data):
     model_names = task_df[['model']]
     return model_names
 
+def format_params(num):
+    if num > 1000000000:
+        if not num % 1000000000:
+            return f'{num // 1000000000}B'
+        return f'{round(num / 1000000000, 1)}B'
+    return f'{num // 1000000}M'
+
 
 demo = gr.Blocks()