From 6457ff33c9023dcb54f882e685306ab09ee0957e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=BB=84=E6=B5=B7?= <10402852@qq.com>
Date: Fri, 28 Jun 2024 10:48:34 +0800
Subject: [PATCH] 'commit'

---
 BaiHu/doTask.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/BaiHu/doTask.py b/BaiHu/doTask.py
index 2b7ca677..827ffc1f 100644
--- a/BaiHu/doTask.py
+++ b/BaiHu/doTask.py
@@ -134,7 +134,6 @@ def download_image(image_url, file_path):
         printf('图片下载失败，状态码：' + str(response.status_code))
 
 
-
 # 指定目录下有多少个指定前缀的文件
 def getPrefixFileCount(directory_path, prefix):
     # 使用os.listdir()获取目录下的所有文件和文件夹
@@ -278,13 +277,13 @@ def get_task():
             if model_type_id == 1:
                 webui_txt_2_img(model_id=model_id, json_data=vjson_data, input_image=input_image)
 
-            # COMFY_UI
-            if model_type_id == 2:
-                runComfyUI(model_id=model_id, json_data=vjson_data, input_image=input_image)
-
             # SD_IMG2IMG
+            if model_type_id == 2:
+                webui_img_2_img(model_id=model_id, prompt_id=prompt_id, json_data=vjson_data,
+                                input_image=input_image)
+            # COMFY_UI
             if model_type_id == 3:
-                webui_img_2_img(model_id=model_id, prompt_id=prompt_id, json_data=vjson_data, input_image=input_image)
+                runComfyUI(model_id=model_id, json_data=vjson_data, input_image=input_image)
 
             return task_id
         elif success == 0: