update code

app.py

@@ -358,8 +358,7 @@ def launch_pretrained():
     )
     logging.info("delete models.")
 
-    os.remove('./pretrained_model/sd21/v2-1_512-ema-pruned.ckpt')
-    os.remove('./pretrained_model/sd21/v2-1_512-nonema-pruned.ckpt')
+
 
     # 下载 CrucibleAI/ControlNetMediaPipeFace 的所有文件
     snapshot_download(
@@ -695,6 +694,7 @@ def style_transfer(processed_image, style_prompt, cfg, strength, save_base,image
     save_dir = os.path.join(save_base, 'style_img')
     os.makedirs(save_dir, exist_ok=True)
     control_image = generate_annotation(src_img_pil, max_faces=1)
+    print(style_prompt)
     trg_img_pil = pipeline_sd(
         prompt=style_prompt,
         image=src_img_pil,