diff --git a/scripts/faceswap.py b/scripts/faceswap.py index afb0cdd..3fce8cb 100644 --- a/scripts/faceswap.py +++ b/scripts/faceswap.py @@ -1,3 +1,4 @@ +import os import gradio as gr import modules.scripts as scripts from modules.upscaler import Upscaler, UpscalerData @@ -18,7 +19,7 @@ import os def get_models(): - models_path = os.path.join(scripts.basedir(), "models/roop/*") + models_path = os.path.join(scripts.basedir(), "models" + os.path.sep + "roop" + os.path.sep + "*") models = glob.glob(models_path) models = [x for x in models if x.endswith(".onnx") or x.endswith(".pth")] return models