Fix path for windows
parent
cdcd31b164
commit
ddc02ee1a9
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue