Temp fix for enhancer

s0md3v · Jul 15, 2023 · ead12e3 · ead12e3
1 parent 9869b96
commit ead12e3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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()