feishen29 commited on
Commit
647f08c
·
verified ·
1 Parent(s): 6f9ef0a

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -170,7 +170,7 @@ def resize_img(input_image, max_side=640, min_side=512, size=None,
170
  @spaces.GPU
171
  def dress_process(garm_img, face_img, pose_img, prompt, cloth_guidance_scale, caption_guidance_scale,
172
  face_guidance_scale,self_guidance_scale, cross_guidance_scale,if_ipa, if_post, if_control, denoise_steps, seed=42):
173
- image_face_fusion = pipeline('face_fusion_torch', model='damo/cv_unet_face_fusion_torch',)
174
  if prompt is None:
175
  prompt = "a photography of a model"
176
  prompt = prompt + ', best quality, high quality'
 
170
  @spaces.GPU
171
  def dress_process(garm_img, face_img, pose_img, prompt, cloth_guidance_scale, caption_guidance_scale,
172
  face_guidance_scale,self_guidance_scale, cross_guidance_scale,if_ipa, if_post, if_control, denoise_steps, seed=42):
173
+ image_face_fusion = pipeline('face_fusion_torch', model='damo/cv_unet_face_fusion_torch',model_revision='v1.0.0')
174
  if prompt is None:
175
  prompt = "a photography of a model"
176
  prompt = prompt + ', best quality, high quality'