Update app.py

app.py

@@ -39,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})'
 
-# --- Leaderboard Table Functions (Modified to sort BEFORE creating HTML bar) ---
+# --- Leaderboard Table Functions (Removed redundant drop) ---
 
 def create_minimal_bar_html(energy_value_wh, energy_score, max_energy_value):
     """Generates HTML for the minimal bar chart with dynamic max energy."""
@@ -74,7 +74,7 @@ def get_model_names(task):
     df['Model'] = df['model'].apply(make_link)
     df['Score'] = df['energy_score'].apply(format_stars)
     df = df[['Model', 'GPU Energy (Wh)', 'Score']] # Keep only these columns
-    df = df.drop('total_gpu_energy', axis=1) # remove the original energy column that was used for sorting
+    # df = df.drop('total_gpu_energy', axis=1) # REMOVED redundant drop
     return df
 
 def get_all_model_names():
@@ -95,7 +95,8 @@ def get_all_model_names():
     all_df['GPU Energy (Wh)'] = all_df.apply(lambda row: create_minimal_bar_html(row['total_gpu_energy'], row['energy_score'], max_energy_overall), axis=1)
     all_df['Model'] = all_df['model'].apply(make_link)
     all_df['Score'] = all_df['energy_score'].apply(format_stars)
-    all_df = all_df.drop('total_gpu_energy', axis=1) # remove the original energy column that was used for sorting
+    all_df = all_df[['Model', 'GPU Energy (Wh)', 'Score']]
+    # all_df = all_df.drop('total_gpu_energy', axis=1) # REMOVED redundant drop
     return all_df[['Model', 'GPU Energy (Wh)', 'Score']]
 
 
@@ -117,7 +118,7 @@ def get_text_generation_model_names(model_class):
     df['Model'] = df['model'].apply(make_link)
     df['Score'] = df['energy_score'].apply(format_stars)
     df = df[['Model', 'GPU Energy (Wh)', 'Score']] # Keep only these columns
-    df = df.drop('total_gpu_energy', axis=1) # remove the original energy column that was used for sorting
+    # df = df.drop('total_gpu_energy', axis=1) # REMOVED redundant drop
     return df
 
 def update_text_generation(model_class):