Lucas Hansen commited on
Commit
cbeba36
1 Parent(s): d7e567f

Update simple.py

Browse files
Files changed (1) hide show
  1. simple.py +1 -1
simple.py CHANGED
@@ -13,7 +13,7 @@ def run_cli(cli_args):
13
  sys.argv = old_argv
14
 
15
  def swap_faces(source_image_path, target_image_path, enhance=True):
16
- provider = 'coreml'
17
 
18
  output_image_file = tempfile.NamedTemporaryFile(suffix='.png')
19
  output_image_path = output_image_file.name
 
13
  sys.argv = old_argv
14
 
15
  def swap_faces(source_image_path, target_image_path, enhance=True):
16
+ provider = 'cuda'
17
 
18
  output_image_file = tempfile.NamedTemporaryFile(suffix='.png')
19
  output_image_path = output_image_file.name