Update processor.py

2023-05-31 02:44:58 +02:00 · 2023-05-31 02:44:58 +02:00 · 6258f3c084
commit 6258f3c084
parent 251fde9596
1 changed files with 1 additions and 1 deletions
--- a/core/processor.py
+++ b/core/processor.py
@ -37,6 +37,6 @@ def process_img(source_img, target_path, output_file):
    frame = cv2.imread(target_path)
    face = get_face(frame)
    source_face = get_face(cv2.imread(source_img))
-    result = face_swapper.get(frame, face, source_face, paste_back=True)
+    result = get_face_swapper().get(frame, face, source_face, paste_back=True)
    cv2.imwrite(output_file, result)
    print("\n\nImage saved as:", output_file, "\n\n")