Fix path for windows

This commit is contained in:
Somdev Sangwan
2023-06-25 08:56:53 +05:30
committed by GitHub
parent cdcd31b164
commit ddc02ee1a9
+2 -1
View File
@@ -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