Upload 6 files

app.py

@@ -0,0 +1,58 @@
+# -* coding:UTF-8 -*
+# !/usr/bin/env python
+import numpy as np
+import roop.globals
+from roop.core import (
+    start,
+    decode_execution_providers,
+    suggest_max_memory,
+    suggest_execution_threads,
+)
+from roop.processors.frame.core import get_frame_processors_modules
+from roop.utilities import normalize_output_path
+from PIL import Image
+import gradio as gr
+import os
+
+def swap_face(source_file, target_file):
+    source_image = Image.fromarray(source_file)
+    source_path = "input.jpg"
+    source_image.save(source_path, format="JPEG", quality=95)
+
+    
+    output_path = "output.mp4"
+
+    roop.globals.source_path = source_path
+    roop.globals.target_path = target_file
+    
+    
+
+    roop.globals.output_path = normalize_output_path(
+        roop.globals.source_path, roop.globals.target_path, output_path
+    )
+    roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
+    roop.globals.headless = True
+    roop.globals.keep_fps = True
+    roop.globals.keep_audio = True
+    roop.globals.keep_frames = False
+    roop.globals.many_faces = False
+    roop.globals.video_encoder = "libx264"
+    roop.globals.video_quality = 50
+    roop.globals.max_memory = suggest_max_memory()
+    roop.globals.execution_providers = decode_execution_providers(["cpu"])
+    roop.globals.execution_threads = suggest_execution_threads()
+    for frame_processor in get_frame_processors_modules(
+        roop.globals.frame_processors
+    ):
+        if not frame_processor.pre_check():
+            return
+
+    start()
+
+    return os.path.join(os.getcwd(), output_path)
+
+app = gr.Interface(
+    fn=swap_face, inputs=[gr.Image(), gr.Video()], outputs=[gr.Video()], description="This model is running on CPU an might be slow. To run in gpu, contact the space owner. To suport the space owner please: https://donate.stripe.com/3csg0D0tadXU4mYcMM"
+)
+
+app.launch()

flake8.txt

@@ -0,0 +1,3 @@
+[flake8]
+select = E3, E4, F
+per-file-ignores = roop/core.py:E402,F401

gitignore.txt

@@ -0,0 +1,4 @@
+.idea
+models
+temp
+__pycache__

mypy.ini

@@ -0,0 +1,7 @@
+[mypy]
+check_untyped_defs = True
+disallow_any_generics = True
+disallow_untyped_calls = True
+disallow_untyped_defs = True
+ignore_missing_imports = True
+strict_optional = False

requirements-headless.txt

@@ -0,0 +1,27 @@
+--extra-index-url https://download.pytorch.org/whl/cu118
+
+numpy==1.23.5
+opencv-python==4.7.0.72
+onnx==1.14.0
+insightface==0.7.3
+psutil==5.9.5
+tk==0.1.0
+customtkinter==5.1.3
+pillow==9.5.0
+torch==2.0.1+cu118; sys_platform != 'darwin'
+torch==2.0.1; sys_platform == 'darwin'
+torchvision==0.15.2+cu118; sys_platform != 'darwin'
+torchvision==0.15.2; sys_platform == 'darwin'
+onnxruntime==1.15.0; sys_platform == 'darwin' and platform_machine != 'arm64'
+onnxruntime-silicon==1.13.1; sys_platform == 'darwin' and platform_machine == 'arm64'
+onnxruntime-gpu==1.15.0; sys_platform != 'darwin'
+tensorflow==2.13.0rc1; sys_platform == 'darwin'
+tensorflow==2.12.0; sys_platform != 'darwin'
+opennsfw2==0.10.2
+protobuf==4.23.2
+tqdm==4.65.0
+gfpgan==1.3.8
+gradio==3.40.1
+tkinterdnd2==0.3.0; sys_platform != 'darwin' and platform_machine != 'arm64'
+tkinterdnd2-universal==1.7.3; sys_platform == 'darwin' and platform_machine == 'arm64'
+onnxruntime-coreml==1.13.1; python_version == '3.9' and sys_platform == 'darwin' and platform_machine != 'arm64'

requirements.txt

@@ -0,0 +1,27 @@
+--extra-index-url https://download.pytorch.org/whl/cu118
+
+numpy==1.23.5
+opencv-python==4.7.0.72
+onnx==1.14.0
+insightface==0.7.3
+psutil==5.9.5
+tk==0.1.0
+customtkinter==5.1.3
+pillow==9.5.0
+torch==2.0.1+cu118; sys_platform != 'darwin'
+torch==2.0.1; sys_platform == 'darwin'
+torchvision==0.15.2+cu118; sys_platform != 'darwin'
+torchvision==0.15.2; sys_platform == 'darwin'
+onnxruntime==1.15.0; sys_platform == 'darwin' and platform_machine != 'arm64'
+onnxruntime-silicon==1.13.1; sys_platform == 'darwin' and platform_machine == 'arm64'
+onnxruntime-gpu==1.15.0; sys_platform != 'darwin'
+tensorflow==2.13.0rc1; sys_platform == 'darwin'
+tensorflow==2.12.0; sys_platform != 'darwin'
+opennsfw2==0.10.2
+protobuf==4.23.2
+tqdm==4.65.0
+gfpgan==1.3.8
+gradio==3.40.1
+tkinterdnd2==0.3.0; sys_platform != 'darwin' and platform_machine != 'arm64'
+tkinterdnd2-universal==1.7.3; sys_platform == 'darwin' and platform_machine == 'arm64'
+onnxruntime-coreml==1.13.1; python_version == '3.9' and sys_platform == 'darwin' and platform_machine != 'arm64'