tuan2308 commited on
Commit
6625b86
1 Parent(s): fdf6664

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -24
app.py CHANGED
@@ -49,20 +49,7 @@ def swap_face(source_file, target_file, doFaceEnhancer):
49
  normalized_output_path = normalize_output_path(source_path, target_path, output_path)
50
 
51
  frame_processors = ["face_swapper", "face_enhancer"] if doFaceEnhancer else ["face_swapper"]
52
- headless = True
53
- keep_fps = True
54
- keep_audio = True
55
- keep_frames = False
56
- many_faces = False
57
- video_encoder = "libx264"
58
- video_quality = 18
59
- max_memory = "12G"
60
- execution_providers = ["CUDAExecutionProvider", "CPUExecutionProvider"] # Ưu tiên GPU, sau đó là CPU
61
- reference_face_position = 0
62
- similar_face_distance = 0.6
63
 
64
- # print("Available providers:", ort.get_available_providers()) # Kiểm tra các provider hiện có
65
- # print("Configured execution providers:", execution_providers)
66
 
67
  for frame_processor in get_frame_processors_modules(frame_processors):
68
  if not frame_processor.pre_check():
@@ -73,17 +60,19 @@ def swap_face(source_file, target_file, doFaceEnhancer):
73
  roop.globals.target_path = target_path
74
  roop.globals.output_path = normalized_output_path
75
  roop.globals.frame_processors = frame_processors
76
- roop.globals.headless = headless
77
- roop.globals.keep_fps = keep_fps
78
- roop.globals.keep_audio = keep_audio
79
- roop.globals.keep_frames = keep_frames
80
- roop.globals.many_faces = many_faces
81
- roop.globals.video_encoder = video_encoder
82
- roop.globals.video_quality = video_quality
83
- roop.globals.execution_providers = execution_providers
84
- roop.globals.reference_face_position = reference_face_position
85
- roop.globals.similar_face_distance = similar_face_distance
86
-
 
 
87
  start()
88
  return normalized_output_path
89
 
 
49
  normalized_output_path = normalize_output_path(source_path, target_path, output_path)
50
 
51
  frame_processors = ["face_swapper", "face_enhancer"] if doFaceEnhancer else ["face_swapper"]
 
 
 
 
 
 
 
 
 
 
 
52
 
 
 
53
 
54
  for frame_processor in get_frame_processors_modules(frame_processors):
55
  if not frame_processor.pre_check():
 
60
  roop.globals.target_path = target_path
61
  roop.globals.output_path = normalized_output_path
62
  roop.globals.frame_processors = frame_processors
63
+ roop.globals.headless = True
64
+ roop.globals.keep_fps = True
65
+ roop.globals.keep_audio = True
66
+ roop.globals.keep_frames = False
67
+ roop.globals.many_faces = False
68
+ roop.globals.video_encoder = "libx264"
69
+ roop.globals.video_quality = 18
70
+ roop.globals.execution_providers = ["CUDAExecutionProvider"]
71
+ roop.globals.reference_face_position = 0
72
+ roop.globals.similar_face_distance = 0.6
73
+ roop.globals.max_memory = 60
74
+ roop.globals.execution_threads = 50
75
+
76
  start()
77
  return normalized_output_path
78