jhj0517 commited on
Commit
f96621b
·
1 Parent(s): e5ef0df

Add `paths`

Browse files
Files changed (1) hide show
  1. modules/utils/paths.py +24 -0
modules/utils/paths.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+
3
+ WEBUI_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".."))
4
+ MODELS_DIR = os.path.join(WEBUI_DIR, "models")
5
+ WHISPER_MODELS_DIR = os.path.join(MODELS_DIR, "Whisper")
6
+ FASTER_WHISPER_MODELS_DIR = os.path.join(WHISPER_MODELS_DIR, "faster-whisper")
7
+ INSANELY_FAST_WHISPER_MODELS_DIR = os.path.join(WHISPER_MODELS_DIR, "insanely-fast-whisper")
8
+ NLLB_MODELS_DIR = os.path.join(MODELS_DIR, "NLLB")
9
+ DIARIZATION_MODELS_DIR = os.path.join(MODELS_DIR, "Diarization")
10
+ CONFIGS_DIR = os.path.join(WEBUI_DIR, "configs")
11
+ DEFAULT_PARAMETERS_PATH = os.path.join(CONFIGS_DIR, "default_parameters.yaml")
12
+ OUTPUT_DIR = os.path.join(WEBUI_DIR, "outputs")
13
+ TRANSLATION_OUTPUT_DIR = os.path.join(OUTPUT_DIR, "translations")
14
+
15
+ for dir_path in [MODELS_DIR,
16
+ WHISPER_MODELS_DIR,
17
+ FASTER_WHISPER_MODELS_DIR,
18
+ INSANELY_FAST_WHISPER_MODELS_DIR,
19
+ NLLB_MODELS_DIR,
20
+ DIARIZATION_MODELS_DIR,
21
+ CONFIGS_DIR,
22
+ OUTPUT_DIR,
23
+ TRANSLATION_OUTPUT_DIR]:
24
+ os.makedirs(dir_path, exist_ok=True)