Hugging Face's logo

Spaces:
ohayonguy
/
PMRF
Running on Zero

App Files Community
4
ohayonguy commited on
Commit
15563ba
•
1 Parent(s): 301c3a1

fixed drawbox

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -100,7 +100,7 @@ def enhance_face(img, face_helper, has_aligned, num_flow_steps, only_center_face
100
 
101
  face_helper.get_inverse_affine(None)
102
  # paste each restored face to the input image
103
- restored_img = face_helper.paste_faces_to_input_image(upsample_img=bg_img, draw_box=False)
104
  return face_helper.cropped_faces, face_helper.restored_faces, restored_img
105
  else:
106
  return face_helper.cropped_faces, face_helper.restored_faces, None
 
100
 
101
  face_helper.get_inverse_affine(None)
102
  # paste each restored face to the input image
103
+ restored_img = face_helper.paste_faces_to_input_image(upsample_img=bg_img)
104
  return face_helper.cropped_faces, face_helper.restored_faces, restored_img
105
  else:
106
  return face_helper.cropped_faces, face_helper.restored_faces, None