File size: 481 Bytes
602a280
 
 
 
 
 
b76be88
602a280
 
 
 
b76be88
 
602a280
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import os

WEBUI_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
MODELS_DIR = os.path.join(WEBUI_DIR, "models")
SAM2_CONFIGS_DIR = os.path.join(WEBUI_DIR, "configs")
OUTPUT_DIR = os.path.join(WEBUI_DIR, "outputs")
OUTPUT_PSD_DIR = os.path.join(OUTPUT_DIR, "psd")

for dir_path in [WEBUI_DIR,
                 MODELS_DIR,
                 SAM2_CONFIGS_DIR,
                 OUTPUT_DIR,
                 OUTPUT_PSD_DIR]:
    os.makedirs(dir_path, exist_ok=True)