diff --git a/core/processor.py b/core/processor.py index a7c5265..535cc17 100644 --- a/core/processor.py +++ b/core/processor.py @@ -1,9 +1,10 @@ import cv2 import insightface +import onnxruntime from core.config import get_face from core.utils import rreplace -face_swapper = insightface.model_zoo.get_model('inswapper_128.onnx', providers=['CUDAExecutionProvider', 'ROCMExecutionProvider', 'CPUExecutionProvider']) +face_swapper = insightface.model_zoo.get_model('inswapper_128.onnx', providers=onnxruntime.get_available_providers()) def process_video(source_img, frame_paths):