Update app.py

app.py

@@ -143,10 +143,13 @@ def plot_heatmap():
     # Add a column for the total scores across all tasks
     df_full["Total Scores"] = df_full.iloc[:, 2:].sum(axis=1)
     
+    # Normalize each column individually for consistent coloring
+    normalized_data = df_full.iloc[:, 2:].apply(lambda x: (x - x.min()) / (x.max() - x.min()), axis=0)
+    
     plt.figure(figsize=(14, 10))
     sns.heatmap(
-        df_full.iloc[:, 2:], 
-        annot=True, 
+        normalized_data, 
+        annot=df_full.iloc[:, 2:],  # Show actual values in annotations
         cmap="YlGnBu", 
         xticklabels=list(columns[2:]) + ["Total Scores"], 
         yticklabels=df_full["Model Configuration"]
@@ -165,6 +168,7 @@ def plot_heatmap():
     return pil_image, temp_image_file.name
 
 
+
 def scrape_mergekit_config(model_name):
     """
     For the *tiny* table’s model links.