Upload app.py

app.py

@@ -164,7 +164,7 @@ def resize_img(input_image, max_side=640, min_side=512, size=None,
 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):
     image_face_fusion = pipeline('face_fusion_torch', model='damo/cv_unet_face_fusion_torch',
-                                 model_revision='v1.0.3', device='cpu')
+                                 model_revision='v1.0.3')
     if prompt is None:
         prompt = "a photography of a model"
     prompt = prompt + ', best quality, high quality'
@@ -257,8 +257,6 @@ face_list = os.listdir(os.path.join(example_path,"face"))
 face_list_path = [os.path.join(example_path,"face",face) for face in face_list]
 
 pose_list = os.listdir(os.path.join(example_path,"pose"))
-print(pose_list)
-print('====', len(pose_list))
 pose_list_path = [os.path.join(example_path,"pose",pose) for pose in pose_list]
 
 def process_image(image):