Update processor.py

core/processor.py

@@ -2,6 +2,7 @@
 
 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")
+    print("\n\nImage saved as:", output_file, "\n\n")