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']
     }
     
     # 任务类型的中文描述