sczhou commited on
Commit
dcc976d
1 Parent(s): 9928d25

add pre_face_align option.

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -207,6 +207,8 @@ def inference(image, face_align, background_enhance, face_upsample, upscale, cod
207
  restored_img = face_helper.paste_faces_to_input_image(
208
  upsample_img=bg_img, draw_box=draw_box
209
  )
 
 
210
 
211
  # save restored img
212
  save_path = f'output/out.png'
 
207
  restored_img = face_helper.paste_faces_to_input_image(
208
  upsample_img=bg_img, draw_box=draw_box
209
  )
210
+ else:
211
+ restored_img = restored_face
212
 
213
  # save restored img
214
  save_path = f'output/out.png'