From c9b76b9673cc5f39f945828feaa9458b87dd3c36 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=A8=E5=8D=9A=E6=96=87?=
 <15549487+FX_YBW@user.noreply.gitee.com>
Date: Wed, 7 Jan 2026 17:59:55 +0800
Subject: [PATCH] =?UTF-8?q?=E8=8E=B7=E5=8F=96=E7=83=AD=E5=8A=9B=E5=9B=BE?=
 =?UTF-8?q?=E9=80=BB=E8=BE=91=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/app/services/image/task_image_strategy.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/backend/app/services/image/task_image_strategy.py b/src/backend/app/services/image/task_image_strategy.py
index ce64065..9267e68 100644
--- a/src/backend/app/services/image/task_image_strategy.py
+++ b/src/backend/app/services/image/task_image_strategy.py
@@ -17,8 +17,8 @@ class TaskImageStrategy:
     TASK_IMAGE_TYPES: Dict[str, List[str]] = {
         'perturbation': ['original', 'perturbed'],
         'finetune': ['original', 'perturbed', 'original_generate', 'perturbed_generate', 'uploaded_generate'],
-        'heatmap': ['report'],
-        'evaluate': ['report'],
+        'heatmap': ['heatmap'],
+        'evaluate': ['report']
     }
     
     # 任务类型的中文描述
-- 
2.34.1