Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import pandas as pd
 import plotly.express as px
+import numpy as np  # Import numpy
 
 CITATION_BUTTON_LABEL = "Copy the following snippet to cite these results"
 CITATION_BUTTON_TEXT = r"""@misc{aienergyscore-leaderboard,
@@ -38,7 +39,7 @@ def make_link(mname):
     display_name = parts[1] if len(parts) > 1 else mname
     return f'[{display_name}](https://huggingface.co/{mname})'
 
-# --- Plot Functions (Bar Chart - Modified) ---
+# --- Plot Functions (Bar Chart - Modified with explicit tickvals) ---
 
 def get_plots(task):
     df = pd.read_csv('data/energy/' + task)
@@ -50,12 +51,6 @@ def get_plots(task):
     # Create a display model column for labeling
     df['Display Model'] = df['model'].apply(lambda m: m.split('/')[-1])
 
-    # --- Debugging: Check data type and dataframe ---
-    print(f"Data type of total_gpu_energy: {df['total_gpu_energy'].dtype}")
-    print("DataFrame head before plotting:")
-    print(df.head())
-    # --- End Debugging ---
-
     # Use the energy score to control color
     color_map = {"1": "red", "2": "orange", "3": "yellow", "4": "lightgreen", "5": "green"}
 
@@ -81,11 +76,10 @@ def get_plots(task):
     fig.update_layout(
         xaxis_title="Model",
         yaxis_title="GPU Energy (Wh)",
-        # --- Removed explicit y-axis formatting for now ---
-        # yaxis_tickformat=".4f",
-        # yaxis = dict(
-        #     tickformat=".4f"
-        # )
+        yaxis = dict(
+            tickformat=".4f",
+            tickvals = list(np.arange(0, df['total_gpu_energy'].max() + 1, 0.5)) # Ticks every 0.5 units
+        )
     )
     return fig
 
@@ -101,13 +95,6 @@ def get_all_plots():
         all_df = pd.concat([all_df, df], ignore_index=True)
     all_df = all_df.drop_duplicates(subset=['model'])
 
-    # --- Debugging: Check data type and dataframe ---
-    print(f"Data type of total_gpu_energy (all_plots): {all_df['total_gpu_energy'].dtype}")
-    print("DataFrame head before plotting (all_plots):")
-    print(all_df.head())
-    # --- End Debugging ---
-
-
     color_map = {"1": "red", "2": "orange", "3": "yellow", "4": "lightgreen", "5": "green"}
 
     fig = px.bar(
@@ -130,15 +117,14 @@ def get_all_plots():
     fig.update_layout(
         xaxis_title="Model",
         yaxis_title="GPU Energy (Wh)",
-        # --- Removed explicit y-axis formatting for now ---
-        # yaxis_tickformat=".4f",
-        # yaxis = dict(
-        #     tickformat=".4f"
-        # )
+        yaxis = dict(
+            tickformat=".4f",
+            tickvals = list(np.arange(0, all_df['total_gpu_energy'].max() + 1, 0.5)) # Ticks every 0.5 units
+        )
     )
     return fig
 
-# --- New functions for Text Generation filtering by model class (with Bar Chart - Modified) ---
+# --- New functions for Text Generation filtering by model class (with Bar Chart - Modified explicit tickvals) ---
 
 def get_text_generation_plots(model_class):
     df = pd.read_csv('data/energy/text_generation.csv')
@@ -151,12 +137,6 @@ def get_text_generation_plots(model_class):
     df['energy_score'] = df['energy_score'].astype(int).astype(str)
     df['Display Model'] = df['model'].apply(lambda m: m.split('/')[-1])
 
-    # --- Debugging: Check data type and dataframe ---
-    print(f"Data type of total_gpu_energy (text_gen): {df['total_gpu_energy'].dtype}")
-    print("DataFrame head before plotting (text_gen):")
-    print(df.head())
-    # --- End Debugging ---
-
 
     color_map = {"1": "red", "2": "orange", "3": "yellow", "4": "lightgreen", "5": "green"}
 
@@ -180,11 +160,10 @@ def get_text_generation_plots(model_class):
     fig.update_layout(
         xaxis_title="Model",
         yaxis_title="GPU Energy (Wh)",
-        # --- Removed explicit y-axis formatting for now ---
-        # yaxis_tickformat=".4f",
-        # yaxis = dict(
-        #     tickformat=".4f"
-        # )
+        yaxis = dict(
+            tickformat=".4f",
+            tickvals = list(np.arange(0, df['total_gpu_energy'].max() + 1, 0.5)) # Ticks every 0.5 units
+        )
     )
     return fig