diff --git a/scripts/faceswap.py b/scripts/faceswap.py index ad79fa6..3ac6cbb 100644 --- a/scripts/faceswap.py +++ b/scripts/faceswap.py @@ -18,8 +18,8 @@ import os def get_models(): - models = glob.glob(models_path) models_path = os.path.join(scripts.basedir(), "models/roop/*") + models = glob.glob(models_path) models += glob.glob(models_path) models = [x for x in models if x.endswith(".onnx") or x.endswith(".pth")] return models