Update app.py

app.py

@@ -3,6 +3,21 @@ import pandas as pd
 from PIL import Image, ImageDraw, ImageFont
 import io
 import os, socket
+
+# --- CSV mapping (HF Space: AIEnergyScore/Leaderboard -> data/energy) ---
+TASK_TO_CSV = {
+    "Image Generation":     "image_generation.csv",
+    "Text Classification":  "text_classification.csv",
+    "Image Classification": "image_classification.csv",
+    "Image Captioning":     "image_captioning.csv",
+    "Summarization":        "summarization.csv",
+    "Text Generation":      "text_generation.csv",
+    "Reasoning":            "reasoning.csv",
+}
+
+# Back-compat so old code like task_to_file[task] keeps working
+task_to_file = TASK_TO_CSV
+
 from huggingface_hub import hf_hub_download
 
 HF_REPO_ID = "AIEnergyScore/Leaderboard"   # this is a Space