Spaces:
Running
on
L4
Running
on
L4
add pre_face_align option.
Browse files
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'
|