Lucas Hansen commited on
Commit
dd4f721
1 Parent(s): 8c613a3

Update simple.py

Browse files
Files changed (1) hide show
  1. simple.py +9 -3
simple.py CHANGED
@@ -12,7 +12,7 @@ def run_cli(cli_args):
12
  finally:
13
  sys.argv = old_argv
14
 
15
- def swap_faces(source_image_path, target_image_path, enhance=True):
16
  provider = 'cuda'
17
 
18
  target_ext = target_image_path.split('.')[-1]
@@ -33,12 +33,17 @@ def swap_faces(source_image_path, target_image_path, enhance=True):
33
  ]
34
 
35
  cli_args += [ '--frame-processors', 'face_swapper' ]
36
- if enhance:
 
37
  cli_args += [
38
- 'frame_enhancer',
39
  'face_enhancer',
40
  ]
41
 
 
 
 
 
 
42
  from facefusion.processors.frame.core import clear_frame_processors_modules
43
  clear_frame_processors_modules()
44
 
@@ -53,6 +58,7 @@ if __name__ == "__main__":
53
  gr.Image(type="filepath"),
54
  gr.Image(type="filepath"),
55
  gr.Checkbox(label="Enhance Face", value=True),
 
56
  ],
57
  outputs=[
58
  gr.Image(
 
12
  finally:
13
  sys.argv = old_argv
14
 
15
+ def swap_faces(source_image_path, target_image_path, enhance_face=True, enhance_frame=True):
16
  provider = 'cuda'
17
 
18
  target_ext = target_image_path.split('.')[-1]
 
33
  ]
34
 
35
  cli_args += [ '--frame-processors', 'face_swapper' ]
36
+
37
+ if enhance_face:
38
  cli_args += [
 
39
  'face_enhancer',
40
  ]
41
 
42
+ if enhance_frame:
43
+ cli_args += [
44
+ 'frame_enhancer',
45
+ ]
46
+
47
  from facefusion.processors.frame.core import clear_frame_processors_modules
48
  clear_frame_processors_modules()
49
 
 
58
  gr.Image(type="filepath"),
59
  gr.Image(type="filepath"),
60
  gr.Checkbox(label="Enhance Face", value=True),
61
+ gr.Checkbox(label="Enhance Frame", value=True),
62
  ],
63
  outputs=[
64
  gr.Image(