Upload 4 files

config_colab.yaml

@@ -0,0 +1,14 @@
+clear_output: true
+force_cpu: false
+max_threads: 3
+memory_limit: 0
+output_image_format: png
+output_template: '{file}_{time}'
+output_video_codec: libx264
+output_video_format: mp4
+provider: cuda
+selected_theme: Default
+server_name: ''
+server_port: 0
+server_share: true
+video_quality: 14

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.txt

@@ -0,0 +1,19 @@
+--extra-index-url https://download.pytorch.org/whl/cu118
+
+numpy==1.24.3
+gradio==4.24.0
+opencv-python==4.8.1.78
+onnx==1.15.0
+insightface==0.7.3
+psutil==5.9.6
+torch==2.1.2+cu118; sys_platform != 'darwin'
+torch==2.1.2; sys_platform == 'darwin'
+torchvision==0.16.2+cu118; sys_platform != 'darwin'
+torchvision==0.16.2; sys_platform == 'darwin'
+onnxruntime==1.16.3; sys_platform == 'darwin' and platform_machine != 'arm64'
+onnxruntime-silicon==1.13.1; sys_platform == 'darwin' and platform_machine == 'arm64'
+onnxruntime-gpu==1.16.3; sys_platform != 'darwin'
+tqdm==4.66.2
+ftfy
+regex
+pyvirtualcam

settings.py

@@ -0,0 +1,68 @@
+import yaml
+
+class Settings:
+    def __init__(self, config_file):
+        self.config_file = config_file
+        self.load()
+
+    def default_get(_, data, name, default):
+        value = default
+        try:
+            value = data.get(name, default)
+        except:
+            pass
+        return value
+
+
+    def load(self):
+        try:
+            with open(self.config_file, 'r') as f:
+                data = yaml.load(f, Loader=yaml.FullLoader)
+        except:
+            data = None
+
+        self.selected_theme = self.default_get(data, 'selected_theme', "Default")
+        self.server_name = self.default_get(data, 'server_name', "")
+        self.server_port = self.default_get(data, 'server_port', 0)
+        self.server_share = self.default_get(data, 'server_share', True)
+        self.output_image_format = self.default_get(data, 'output_image_format', 'png')
+        self.output_video_format = self.default_get(data, 'output_video_format', 'mp4')
+        self.output_video_codec = self.default_get(data, 'output_video_codec', 'libx264')
+        self.video_quality = self.default_get(data, 'video_quality', 14)
+        self.clear_output = self.default_get(data, 'clear_output', True)
+        self.max_threads = self.default_get(data, 'max_threads', 2)
+        self.memory_limit = self.default_get(data, 'memory_limit', 0)
+        self.provider = self.default_get(data, 'provider', 'cuda')
+        self.force_cpu = self.default_get(data, 'force_cpu', False)
+        self.output_template = self.default_get(data, 'output_template', '{file}_{time}')
+        self.use_os_temp_folder = self.default_get(data, 'use_os_temp_folder', False)
+        self.output_show_video = self.default_get(data, 'output_show_video', True)
+
+
+
+
+
+    def save(self):
+        data = {
+            'selected_theme': self.selected_theme,
+            'server_name': self.server_name,
+            'server_port': self.server_port,
+            'server_share': self.server_share,
+            'output_image_format' : self.output_image_format,
+            'output_video_format' : self.output_video_format,
+            'output_video_codec' : self.output_video_codec,
+            'video_quality' : self.video_quality,
+            'clear_output' : self.clear_output,
+            'max_threads' : self.max_threads,
+            'memory_limit' : self.memory_limit,
+            'provider' : self.provider,
+            'force_cpu' : self.force_cpu,
+			'output_template' : self.output_template,
+            'use_os_temp_folder' : self.use_os_temp_folder,
+            'output_show_video' : self.output_show_video
+        }
+        with open(self.config_file, 'w') as f:
+            yaml.dump(data, f)
+
+
+