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'
|