Update app.py

app.py

@@ -32,8 +32,8 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     target_image = Image.fromarray(target_file)
     target_image.save(target_path)
 
-    # print("source_path: ", source_path)
-    # print("target_path: ", target_path)
+    print("source_path: ", source_path)
+    print("target_path: ", target_path)
 
     # Check if a face is detected in the source image
     source_face = get_one_face(cv2.imread(source_path))
@@ -49,20 +49,7 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     normalized_output_path = normalize_output_path(source_path, target_path, output_path)
 
     frame_processors = ["face_swapper", "face_enhancer"] if doFaceEnhancer else ["face_swapper"]
-    headless = True
-    keep_fps = True
-    keep_audio = True
-    keep_frames = False
-    many_faces = False
-    video_encoder = "libx264"
-    video_quality = 18
-    max_memory = "12G"
-    execution_providers = ["CUDAExecutionProvider", "CPUExecutionProvider"]  # Ưu tiên GPU, sau đó là CPU
-    reference_face_position = 0
-    similar_face_distance = 0.6
 
-    # print("Available providers:", ort.get_available_providers())  # Kiểm tra các provider hiện có
-    # print("Configured execution providers:", execution_providers)
 
     for frame_processor in get_frame_processors_modules(frame_processors):
         if not frame_processor.pre_check():
@@ -73,17 +60,19 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     roop.globals.target_path = target_path
     roop.globals.output_path = normalized_output_path
     roop.globals.frame_processors = frame_processors
-    roop.globals.headless = headless
-    roop.globals.keep_fps = keep_fps
-    roop.globals.keep_audio = keep_audio
-    roop.globals.keep_frames = keep_frames
-    roop.globals.many_faces = many_faces
-    roop.globals.video_encoder = video_encoder
-    roop.globals.video_quality = video_quality
-    roop.globals.execution_providers = execution_providers
-    roop.globals.reference_face_position = reference_face_position
-    roop.globals.similar_face_distance = similar_face_distance
-
+    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.execution_providers = ["CUDAExecutionProvider"]
+    roop.globals.reference_face_position = 0
+    roop.globals.similar_face_distance = 0.6
+    roop.globals.max_memory = 60
+    roop.globals.execution_threads = 50
+    
     start()
     return normalized_output_path
 
@@ -96,4 +85,4 @@ app = gr.Interface(
     ], 
     outputs="image"
 )
-app.launch()
+app.launch()