diff --git a/core/processor.py b/core/processor.py index 26cb3df..b8ec133 100644 --- a/core/processor.py +++ b/core/processor.py @@ -2,6 +2,7 @@ import os import cv2 import insightface +import core.globals from core.config import get_face FACE_SWAPPER = None @@ -11,7 +12,7 @@ def get_face_swapper(): global FACE_SWAPPER if FACE_SWAPPER is None: model_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), '../inswapper_128.onnx') - FACE_SWAPPER = insightface.model_zoo.get_model(model_path) + FACE_SWAPPER = insightface.model_zoo.get_model(model_path, providers=core.globals.providers) return FACE_SWAPPER @@ -38,4 +39,4 @@ def process_img(source_img, target_path, output_file): source_face = get_face(cv2.imread(source_img)) 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") \ No newline at end of file + print("\n\nImage saved as:", output_file, "\n\n")