diff --git a/utils.py b/utils.py index d4faef1..f8f9fe7 100644 --- a/utils.py +++ b/utils.py @@ -62,7 +62,7 @@ class Models: @classmethod def lmk(cls, crop): if '_lmk' not in cls.__dict__: - cls._lmk = InferenceSession(os.path.join(models_dir, 'landmarks', 'fan2_68_landmark.onnx'), providers=get_available_providers()) + cls._lmk = InferenceSession(os.path.join(models_dir, 'landmarks', 'fan2_68_landmark.onnx')) lmk = cls._lmk.run(None, {'input': crop})[0] return lmk