Update app.py

app.py

@@ -59,7 +59,7 @@ def get_plots(task):
         width=800,
         color_discrete_map=color_map
     )
-    # GPU Energy now shows 4 decimals in the hover text
+    # Hover text now shows GPU Energy (Wh) with 4 decimals
     fig.update_traces(
         hovertemplate="<br>".join([
             "Model: %{y}",
@@ -67,7 +67,12 @@ def get_plots(task):
             "Energy Score: %{customdata[0]}"
         ])
     )
-    fig.update_layout(xaxis_title="GPU Energy (Wh)", yaxis_title="Model")
+    # Set the x-axis tickformat to 4 decimals
+    fig.update_layout(
+        xaxis_title="GPU Energy (Wh)",
+        yaxis_title="Model",
+        xaxis=dict(tickformat=".4f")
+    )
     return fig
 
 def get_all_plots():
@@ -101,7 +106,11 @@ def get_all_plots():
             "Energy Score: %{customdata[0]}"
         ])
     )
-    fig.update_layout(xaxis_title="GPU Energy (Wh)", yaxis_title="Model")
+    fig.update_layout(
+        xaxis_title="GPU Energy (Wh)",
+        yaxis_title="Model",
+        xaxis=dict(tickformat=".4f")
+    )
     return fig
 
 def get_model_names(task):
@@ -109,11 +118,11 @@ def get_model_names(task):
     if df.columns[0].startswith("Unnamed:"):
         df = df.iloc[:, 1:]
     df['energy_score'] = df['energy_score'].astype(int)
-    # Ensure GPU Energy is a float and format it to 4 decimals
+    # Ensure GPU Energy is a float and format it to 4 decimals for display
     df['GPU Energy (Wh)'] = df['total_gpu_energy'].astype(float).apply(lambda x: f"{x:.4f}")
     df['Model'] = df['model'].apply(make_link)
     df['Score'] = df['energy_score'].apply(format_stars)
-    # Remove any Class column
+    # Remove any Class column if it exists
     df = df[['Model', 'GPU Energy (Wh)', 'Score']]
     df = df.sort_values(by='GPU Energy (Wh)')
     return df
@@ -162,7 +171,11 @@ def get_text_generation_plots(model_class):
             "Energy Score: %{customdata[0]}"
         ])
     )
-    fig.update_layout(xaxis_title="GPU Energy (Wh)", yaxis_title="Model")
+    fig.update_layout(
+        xaxis_title="GPU Energy (Wh)",
+        yaxis_title="Model",
+        xaxis=dict(tickformat=".4f")
+    )
     return fig
 
 def get_text_generation_model_names(model_class):