LosCaquitos commited on
Commit
f829ba3
Β·
verified Β·
1 Parent(s): e4f2993

Update lib/config.py

Browse files
Files changed (1) hide show
  1. lib/config.py +6 -6
lib/config.py CHANGED
@@ -37,20 +37,20 @@ BUILTIN_MODELS = {
37
  "Gawr Gura": "gawr_gura",
38
  }
39
 
40
- # ── Demucs configuration (new) ──────────────────────────────────────────────
41
- DEMUCS_MODEL = "htdemucs" # best quality, also supports "--two-stems=vocals"
42
  DEMUCS_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
43
- DEMUCS_SEPARATION_DIR = "separated" # temporary directory for Demucs output
44
 
45
  # ── Output formats ──────────────────────────────────────────────────────────
46
  OUTPUT_FORMATS = ["WAV", "MP3", "FLAC", "OPUS"]
47
- DEFAULT_OUTPUT_FORMAT = "MP3" # good balance for web
48
 
49
  # ── Paths ───────────────────────────────────────────────────────────────────
50
  MODELS_DIR = "models"
51
- OUTPUTS_DIR = "outputs" # new: persistent folder for the 5 files
52
 
53
- # ── Gradio CSS (original) ───────────────────────────────────────────────────
54
  CSS = """
55
  #header { text-align: center; margin-bottom: 1rem; }
56
  #status { text-align: center; font-weight: bold; }
 
37
  "Gawr Gura": "gawr_gura",
38
  }
39
 
40
+ # ── Demucs configuration ────────────────────────────────────────────────────
41
+ DEMUCS_MODEL = "htdemucs"
42
  DEMUCS_DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
43
+ DEMUCS_SEPARATION_DIR = "separated"
44
 
45
  # ── Output formats ──────────────────────────────────────────────────────────
46
  OUTPUT_FORMATS = ["WAV", "MP3", "FLAC", "OPUS"]
47
+ DEFAULT_OUTPUT_FORMAT = "MP3"
48
 
49
  # ── Paths ───────────────────────────────────────────────────────────────────
50
  MODELS_DIR = "models"
51
+ OUTPUTS_DIR = "outputs"
52
 
53
+ # ── Gradio CSS ──────────────────────────────────────────────────────────────
54
  CSS = """
55
  #header { text-align: center; margin-bottom: 1rem; }
56
  #status { text-align: center; font-weight: bold; }