Update app.py

app.py

@@ -6,8 +6,8 @@ import roop.globals
 from roop.core import (
     start,
     decode_execution_providers,
-    suggest_max_memory,
-    suggest_execution_threads,
+    # suggest_max_memory,  # Loại bỏ hàm này
+    # suggest_execution_threads,  # Loại bỏ hàm này
 )
 from roop.processors.frame.core import get_frame_processors_modules
 from roop.utilities import normalize_output_path
@@ -16,7 +16,7 @@ from PIL import Image
 import spaces
 
 @spaces.GPU
-def swap_face(source_file, target_file,doFaceEnhancer):
+def swap_face(source_file, target_file, doFaceEnhancer):
 
     source_path = "input.jpg"
     target_path = "target.jpg"
@@ -36,7 +36,7 @@ def swap_face(source_file, target_file,doFaceEnhancer):
         roop.globals.source_path, roop.globals.target_path, output_path
     )
     if doFaceEnhancer == True:
-        roop.globals.frame_processors = ["face_swapper","face_enhancer"]
+        roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
     else:
         roop.globals.frame_processors = ["face_swapper"]
     roop.globals.headless = True
@@ -46,9 +46,9 @@ def swap_face(source_file, target_file,doFaceEnhancer):
     roop.globals.many_faces = False
     roop.globals.video_encoder = "libx264"
     roop.globals.video_quality = 18
-    roop.globals.max_memory = suggest_max_memory()
+    roop.globals.max_memory = "4G"  # Sử dụng giá trị mặc định
     roop.globals.execution_providers = decode_execution_providers(["cuda"])
-    roop.globals.execution_threads = suggest_execution_threads()
+    roop.globals.execution_threads = 4  # Sử dụng giá trị mặc định
 
     print(
         "start process",
@@ -66,8 +66,7 @@ def swap_face(source_file, target_file,doFaceEnhancer):
     start()
     return output_path
 
-
 app = gr.Interface(
-    fn=swap_face, inputs=[gr.Image(), gr.Image(),gr.Checkbox(label="face_enhancer?", info="do face enhancer?")], outputs="image"
+    fn=swap_face, inputs=[gr.Image(), gr.Image(), gr.Checkbox(label="face_enhancer?", info="do face enhancer?")], outputs="image"
 )
-app.launch()
+app.launch()