jhj0517
Apply constants
7d9eec3
raw
history blame
1.11 kB
import os
WEBUI_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".."))
MODELS_DIR = os.path.join(WEBUI_DIR, "models")
WHISPER_MODELS_DIR = os.path.join(MODELS_DIR, "Whisper")
FASTER_WHISPER_MODELS_DIR = os.path.join(WHISPER_MODELS_DIR, "faster-whisper")
INSANELY_FAST_WHISPER_MODELS_DIR = os.path.join(WHISPER_MODELS_DIR, "insanely-fast-whisper")
NLLB_MODELS_DIR = os.path.join(MODELS_DIR, "NLLB")
DIARIZATION_MODELS_DIR = os.path.join(MODELS_DIR, "Diarization")
CONFIGS_DIR = os.path.join(WEBUI_DIR, "configs")
DEFAULT_PARAMETERS_CONFIG_PATH = os.path.join(CONFIGS_DIR, "default_parameters.yaml")
OUTPUT_DIR = os.path.join(WEBUI_DIR, "outputs")
TRANSLATION_OUTPUT_DIR = os.path.join(OUTPUT_DIR, "translations")
for dir_path in [MODELS_DIR,
WHISPER_MODELS_DIR,
FASTER_WHISPER_MODELS_DIR,
INSANELY_FAST_WHISPER_MODELS_DIR,
NLLB_MODELS_DIR,
DIARIZATION_MODELS_DIR,
CONFIGS_DIR,
OUTPUT_DIR,
TRANSLATION_OUTPUT_DIR]:
os.makedirs(dir_path, exist_ok=True)