Create simple.py

simple.py

@@ -0,0 +1,65 @@
+import gradio as gr
+from PIL import Image
+import sys
+import tempfile
+from facefusion import core
+
+def run_cli(cli_args):
+  old_argv = sys.argv
+  try:
+    sys.argv = ['run.py', *cli_args]
+    core.cli()
+  finally:
+    sys.argv = old_argv
+
+def swap_faces(source_image_path, target_image_path, enhance=True):
+  provider = 'coreml'
+
+  output_image_file = tempfile.NamedTemporaryFile(suffix='.png')
+  output_image_path = output_image_file.name
+
+  print(source_image_path)
+  print(target_image_path)
+  print(output_image_path)
+
+  cli_args = [
+    '--headless',
+    '-s', source_image_path,
+    '-t', target_image_path,
+    '-o', output_image_path,
+    '--output-image-quality', '80',
+    '--execution-providers', provider,
+  ]
+
+  cli_args += [ '--frame-processors', 'face_swapper' ]
+  if enhance:
+    cli_args += [
+      'frame_enhancer',
+      'face_enhancer',
+    ]
+
+  from facefusion.processors.frame.core import clear_frame_processors_modules
+  clear_frame_processors_modules()
+
+  run_cli(cli_args)
+
+  return Image.open(output_image_path)
+
+if __name__ == "__main__":
+    demo = gr.Interface(
+      fn=swap_faces,
+      inputs=[
+          gr.Image(type="filepath"),
+          gr.Image(type="filepath"),
+          gr.Checkbox(label="Enhance Face", value=True),
+      ],
+      outputs=[
+          gr.Image(
+            type="pil",
+            show_download_button=True,
+          )
+      ],
+      title="Swap Faces",
+      allow_flagging="never"
+    )
+    demo.launch()