Upload app.py

app.py

@@ -8,7 +8,7 @@ from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
 from dressing_sd.pipelines.pipeline_sd import PipIpaControlNet
 from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
-
+import spaces
 from torchvision import transforms
 import cv2
 from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
@@ -89,7 +89,7 @@ unet = UNet2DConditionModel.from_pretrained("./ckpt/unet").to(
 # image_face_fusion = pipeline('face_fusion_torch', model='damo/cv_unet_face_fusion_torch', model_revision='v1.0.3')
 
 #face_model
-app = FaceAnalysis(providers=[('CUDAExecutionProvider', {"device_id": args.device})]) ##使用GPU:0, 默认使用buffalo_l就可以了
+app = FaceAnalysis(model_path='./ckpt/buffalo_l.zip', providers=[('CUDAExecutionProvider', {"device_id": args.device})]) ##使用GPU:0, 默认使用buffalo_l就可以了
 app.prepare(ctx_id=0, det_size=(640, 640))
 
 # def ref proj weight
@@ -201,7 +201,7 @@ def resize_img(input_image, max_side=640, min_side=512, size=None,
     return input_image
 
 @spaces.GPU
-def tryon_process(garm_img, face_img, pose_img, prompt, cloth_guidance_scale, caption_guidance_scale,
+def dress_process(garm_img, face_img, pose_img, prompt, cloth_guidance_scale, caption_guidance_scale,
                   face_guidance_scale,self_guidance_scale, cross_guidance_scale,if_ipa, if_post, if_control, denoise_steps, seed=42):
     # prompt = prompt + ', confident smile expression, fashion, best quality, amazing quality, very aesthetic'
     if prompt is None:
@@ -394,7 +394,7 @@ with image_blocks as demo:
                 denoise_steps = gr.Number(label="Denoising Steps", minimum=20, maximum=50, value=30, step=1)
                 seed = gr.Number(label="Seed", minimum=-1, maximum=2147483647, step=1, value=20240508)
     
-    try_button.click(fn=tryon_process, inputs=[garm_img, imgs, pose_img, prompt, cloth_guidance_scale, caption_guidance_scale, face_guidance_scale,self_guidance_scale, cross_guidance_scale, is_checked_face, is_checked_postprocess, is_checked_pose, denoise_steps, seed],
-                     outputs=[image_out], api_name='tryon')
+    try_button.click(fn=dress_process, inputs=[garm_img, imgs, pose_img, prompt, cloth_guidance_scale, caption_guidance_scale, face_guidance_scale,self_guidance_scale, cross_guidance_scale, is_checked_face, is_checked_postprocess, is_checked_pose, denoise_steps, seed],
+                     outputs=[image_out], api_name='IMAGDressing-v1')
 
 image_blocks.launch()