update code

app.py

@@ -648,6 +648,7 @@ def assert_input_image(input_image):
 @spaces.GPU(duration=30)
 def process_image(input_image_dir, source_type, is_style, save_dir):
         
+        
         """ 🎯 处理 input_image，根据是否是示例图片执行不同逻辑 """
         process_img_input_dir = os.path.join(save_dir, 'input_image')
         process_img_save_dir = os.path.join(save_dir, 'processed_img')
@@ -658,7 +659,7 @@ def process_image(input_image_dir, source_type, is_style, save_dir):
         os.makedirs(process_img_input_dir, exist_ok=True)
         if source_type == "example":
             image = Image.open(input_image_dir)
-            return image, source_type, image_name_true, ""
+            return image, source_type, image_name_true
         else:
             # input_process_model.inference(input_image, process_img_save_dir)
             shutil.copy(input_image_dir, process_img_input_dir)
@@ -672,7 +673,7 @@ def process_image(input_image_dir, source_type, is_style, save_dir):
             imge_dir = os.path.join(process_img_save_dir, 'dataset/images512x512/input_image', closest_match)
             image = Image.open(imge_dir)
             image_name_true = closest_match
-            return image, source_type, image_name_true, ""  # 这里替换成 处理用户上传图片的逻辑
+            return image, source_type, image_name_true  # 这里替换成 处理用户上传图片的逻辑
 
 
 
@@ -699,7 +700,7 @@ def style_transfer(processed_image, style_prompt, cfg, strength, save_base,image
         controlnet_conditioning_scale=1.5
     )['images'][0]
     trg_img_pil.save(os.path.join(save_dir, image_name_true))
-    return trg_img_pil, "" # 🚨 这里需要替换成你的风格转换逻辑
+    return trg_img_pil # 🚨 这里需要替换成你的风格转换逻辑