diff --git a/api/tasks/image_generation_task.py b/api/tasks/image_generation_task.py
index 5b3a6ef340..c315c29867 100644
--- a/api/tasks/image_generation_task.py
+++ b/api/tasks/image_generation_task.py
@@ -4,6 +4,7 @@ import time
 
 import click
 from celery import shared_task  # type: ignore
+
 from configs import dify_config
 from core.app.entities.app_invoke_entities import InvokeFrom
 from extensions.ext_database import db