diff --git a/roop/processors/frame/face_enhancer.py b/roop/processors/frame/face_enhancer.py index 63d32a41a..a4c9a3a07 100644 --- a/roop/processors/frame/face_enhancer.py +++ b/roop/processors/frame/face_enhancer.py @@ -67,7 +67,7 @@ def enhance_face(temp_frame: Frame) -> Frame: return temp_frame -def process_frame(source_face: Face, temp_frame: Frame) -> Frame: +def process_frame(source_face: Face, reference_face: Face, temp_frame: Frame) -> Frame: target_face = get_one_face(temp_frame) if target_face: temp_frame = enhance_face(temp_frame) @@ -77,7 +77,7 @@ def process_frame(source_face: Face, temp_frame: Frame) -> Frame: def process_frames(source_path: str, temp_frame_paths: List[str], update: Callable[[], None]) -> None: for temp_frame_path in temp_frame_paths: temp_frame = cv2.imread(temp_frame_path) - result = process_frame(None, temp_frame) + result = process_frame(None, None, temp_frame) cv2.imwrite(temp_frame_path, result) if update: update()