Update app.py

app.py

@@ -140,12 +140,15 @@ def plot_task_specific_top_models():
     return pil_image, temp_image_file.name
 
 def plot_heatmap():
+    # Add a column for the total scores across all tasks
+    df_full["Total Scores"] = df_full.iloc[:, 2:].sum(axis=1)
+    
     plt.figure(figsize=(14, 10))
     sns.heatmap(
         df_full.iloc[:, 2:], 
         annot=True, 
         cmap="YlGnBu", 
-        xticklabels=columns[2:], 
+        xticklabels=list(columns[2:]) + ["Total Scores"], 
         yticklabels=df_full["Model Configuration"]
     )
     plt.title("Performance Heatmap", fontsize=16)
@@ -161,6 +164,7 @@ def plot_heatmap():
     pil_image.save(temp_image_file.name)
     return pil_image, temp_image_file.name
 
+
 def scrape_mergekit_config(model_name):
     """
     For the *tiny* table’s model links.