memasukkan roop

app.py

@@ -1,18 +1,20 @@
 import numpy as np
 import gradio as gr
-import models_client.globals
-from models_client.core import (
+import roop.globals
+from roop.core import (
     start,
     decode_execution_providers,
     suggest_max_memory,
     suggest_execution_threads,
 )
-from models_client.processors.frame.core import get_frame_processors_modules
-from models_client.utilities import normalize_output_path
+from roop.processors.frame.core import get_frame_processors_modules
+from roop.utilities import normalize_output_path
 import os
 from PIL import Image
 
+
 def swap_face(source_file, target_file, doFaceEnhancer):
+
     source_path = "input.jpg"
     target_path = "target.jpg"
 
@@ -24,36 +26,36 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     print("source_path: ", source_path)
     print("target_path: ", target_path)
 
-    models_client.globals.source_path = source_path
-    models_client.globals.target_path = target_path
+    roop.globals.source_path = source_path
+    roop.globals.target_path = target_path
     output_path = "output.jpg"
-    models_client.globals.output_path = normalize_output_path(
-        models_client.globals.source_path, models_client.globals.target_path, output_path
+    roop.globals.output_path = normalize_output_path(
+        roop.globals.source_path, roop.globals.target_path, output_path
     )
     if doFaceEnhancer:
-        models_client.globals.frame_processors = ["face_swapper", "face_enhancer"]
+        roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
     else:
-        models_client.globals.frame_processors = ["face_swapper"]
-    models_client.globals.headless = True
-    models_client.globals.keep_fps = True
-    models_client.globals.keep_audio = True
-    models_client.globals.keep_frames = False
-    models_client.globals.many_faces = False
-    models_client.globals.video_encoder = "libx264"
-    models_client.globals.video_quality = 18
-    models_client.globals.max_memory = suggest_max_memory()
-    models_client.globals.execution_providers = decode_execution_providers(["cuda"])
-    models_client.globals.execution_threads = suggest_execution_threads()
+        roop.globals.frame_processors = ["face_swapper"]
+    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 = 18
+    roop.globals.max_memory = suggest_max_memory()
+    roop.globals.execution_providers = decode_execution_providers(["cuda"])
+    roop.globals.execution_threads = suggest_execution_threads()
 
     print(
         "start process",
-        models_client.globals.source_path,
-        models_client.globals.target_path,
-        models_client.globals.output_path,
+        roop.globals.source_path,
+        roop.globals.target_path,
+        roop.globals.output_path,
     )
 
     for frame_processor in get_frame_processors_modules(
-        models_client.globals.frame_processors
+        roop.globals.frame_processors
     ):
         if not frame_processor.pre_check():
             return
@@ -61,8 +63,9 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     start()
     return output_path
 
-html_section_1 = "<div style='text:align:center'><h1>Ai Penganti Wajah</h1></div>"
-html_section_2 = "<div style='text:align:center'><p>Unggah gambar sumber dan target Anda untuk bertukar wajah, Kemudian anda bisa meningkatkan kualitas wajah dengan cara mencentang opsi dibawah</p></div>"
+
+html_section_1 = "<div style="text:align:center"><h1>Ai Penganti Wajah</h1></div>"
+html_section_2 = "<div style="text:align:center"><p>Unggah gambar sumber dan target Anda untuk bertukar wajah, Kemudian anda bisa meningkatkan kualitas wajah dengan cara mencentang opsi dibawah</p></div>"
 
 app = gr.Blocks()