diff --git a/blueprints/api.py b/blueprints/api.py
index 1c2e399..e121013 100644
--- a/blueprints/api.py
+++ b/blueprints/api.py
@@ -72,7 +72,8 @@ def generate():
 
         model_value = data.get('model')
         prompt = data.get('prompt')
-        is_chat_model = "gemini" in model_value.lower() or "gpt" in model_value.lower()
+        model_lower = model_value.lower()
+        is_chat_model = ("gemini" in model_lower or "gpt" in model_lower) and "image" not in model_lower
 
         # 3. 处理聊天模型 (同步)
         if is_chat_model: