Politrees commited on
Commit
f018f0f
·
verified ·
1 Parent(s): 2f58437

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -18
app.py CHANGED
@@ -144,13 +144,13 @@ def prepare_output_dir(input_file, output_dir):
144
  def rename_stems(audio, vocals_stem, instrumental_stem, other_stem, drums_stem, bass_stem, guitar_stem, piano_stem, model):
145
  base_name = os.path.splitext(os.path.basename(audio))[0]
146
  stems = {
147
- "Vocals": vocals_stem.replace("<IN>", base_name).replace("<SN>", "Vocals").replace("<MN>", model),
148
- "Instrumental": instrumental_stem.replace("<IN>", base_name).replace("<SN>", "Instrumental").replace("<MN>", model),
149
- "Drums": drums_stem.replace("<IN>", base_name).replace("<SN>", "Drums").replace("<MN>", model),
150
- "Bass": bass_stem.replace("<IN>", base_name).replace("<SN>", "Bass").replace("<MN>", model),
151
- "Other": other_stem.replace("<IN>", base_name).replace("<SN>", "Other").replace("<MN>", model),
152
- "Guitar": guitar_stem.replace("<IN>", base_name).replace("<SN>", "Guitar").replace("<MN>", model),
153
- "Piano": piano_stem.replace("<IN>", base_name).replace("<SN>", "Piano").replace("<MN>", model),
154
  }
155
  return stems
156
 
@@ -460,21 +460,21 @@ with gr.Blocks(
460
  batch_size = gr.Slider(minimum=1, maximum=16, step=1, value=1, label="Batch Size", info="Larger consumes more RAM but may process slightly faster.")
461
 
462
  with gr.Accordion("Rename Stems", open=False):
463
- gr.HTML("<h3> Input File Name: <IN> - Automatic detection </h3>")
464
- gr.HTML("<h3> Stem Name: <SN> - Automatic detection (e.g., Vocals, Instrumental) </h3>")
465
- gr.HTML("<h3> Model Name: <MN> - Automatic detection (e.g., BS-Roformer-Viperx-1297) </h3>")
466
- gr.HTML("<h3> Usage Example: <IN>_<SN>_<MN> </h3>")
467
  gr.HTML("<h3> Output File Name: Music_(Vocals)_BS-Roformer-Viperx-1297 </h3>")
468
  with gr.Row():
469
- vocals_stem = gr.Textbox(value="<IN>_(<SN>)_<MN>", label="Vocals Stem", info="Пример вывода: Music_(Vocals)_BS-Roformer-Viperx-1297", placeholder="<IN>_(<SN>)_<MN>")
470
- instrumental_stem = gr.Textbox(value="<IN>_(<SN>)_<MN>", label="Instrumental Stem", info="Пример вывода: Music_(Instrumental)_BS-Roformer-Viperx-1297", placeholder="<IN>_(<SN>)_<MN>")
471
- other_stem = gr.Textbox(value="<IN>_(<SN>)_<MN>", label="Other Stem", info="Пример вывода: Music_(Other)_BS-Roformer-Viperx-1297", placeholder="<IN>_(<SN>)_<MN>")
472
  with gr.Row():
473
- drums_stem = gr.Textbox(value="<IN>_(<SN>)_<MN>", label="Drums Stem", info="Пример вывода: Music_(Drums)_BS-Roformer-Viperx-1297", placeholder="<IN>_(<SN>)_<MN>")
474
- bass_stem = gr.Textbox(value="<IN>_(<SN>)_<MN>", label="Bass Stem", info="Пример вывода: Music_(Bass)_BS-Roformer-Viperx-1297", placeholder="<IN>_(<SN>)_<MN>")
475
  with gr.Row():
476
- guitar_stem = gr.Textbox(value="<IN>_(<SN>)_<MN>", label="Guitar Stem", info="Пример вывода: Music_(Guitar)_BS-Roformer-Viperx-1297", placeholder="<IN>_(<SN>)_<MN>")
477
- piano_stem = gr.Textbox(value="<IN>_(<SN>)_<MN>", label="Piano Stem", info="Пример вывода: Music_(Piano)_BS-Roformer-Viperx-1297", placeholder="<IN>_(<SN>)_<MN>")
478
 
479
  demucs_model.change(update_stems, inputs=[demucs_model], outputs=stem6)
480
 
 
144
  def rename_stems(audio, vocals_stem, instrumental_stem, other_stem, drums_stem, bass_stem, guitar_stem, piano_stem, model):
145
  base_name = os.path.splitext(os.path.basename(audio))[0]
146
  stems = {
147
+ "Vocals": vocals_stem.replace("NAME", base_name).replace("STEM", "Vocals").replace("MODEL", model),
148
+ "Instrumental": instrumental_stem.replace("NAME", base_name).replace("STEM", "Instrumental").replace("MODEL", model),
149
+ "Drums": drums_stem.replace("NAME", base_name).replace("STEM", "Drums").replace("MODEL", model),
150
+ "Bass": bass_stem.replace("NAME", base_name).replace("STEM", "Bass").replace("MODEL", model),
151
+ "Other": other_stem.replace("NAME", base_name).replace("STEM", "Other").replace("MODEL", model),
152
+ "Guitar": guitar_stem.replace("NAME", base_name).replace("STEM", "Guitar").replace("MODEL", model),
153
+ "Piano": piano_stem.replace("NAME", base_name).replace("STEM", "Piano").replace("MODEL", model),
154
  }
155
  return stems
156
 
 
460
  batch_size = gr.Slider(minimum=1, maximum=16, step=1, value=1, label="Batch Size", info="Larger consumes more RAM but may process slightly faster.")
461
 
462
  with gr.Accordion("Rename Stems", open=False):
463
+ gr.HTML("<h3> NAME - Input File Name </h3>")
464
+ gr.HTML("<h3> STEM - Stem Name (e.g., Vocals, Instrumental) </h3>")
465
+ gr.HTML("<h3> MODEL - Model Name (e.g., BS-Roformer-Viperx-1297) </h3>")
466
+ gr.HTML("<h3> Usage Example: NAME_(STEM)_MODEL </h3>")
467
  gr.HTML("<h3> Output File Name: Music_(Vocals)_BS-Roformer-Viperx-1297 </h3>")
468
  with gr.Row():
469
+ vocals_stem = gr.Textbox(value="NAME_(STEM)_MODEL", label="Vocals Stem", info="Пример вывода: Music_(Vocals)_BS-Roformer-Viperx-1297", placeholder="NAME_(STEM)_MODEL")
470
+ instrumental_stem = gr.Textbox(value="NAME_(STEM)_MODEL", label="Instrumental Stem", info="Пример вывода: Music_(Instrumental)_BS-Roformer-Viperx-1297", placeholder="NAME_(STEM)_MODEL")
471
+ other_stem = gr.Textbox(value="NAME_(STEM)_MODEL", label="Other Stem", info="Пример вывода: Music_(Other)_BS-Roformer-Viperx-1297", placeholder="NAME_(STEM)_MODEL")
472
  with gr.Row():
473
+ drums_stem = gr.Textbox(value="NAME_(STEM)_MODEL", label="Drums Stem", info="Пример вывода: Music_(Drums)_BS-Roformer-Viperx-1297", placeholder="NAME_(STEM)_MODEL")
474
+ bass_stem = gr.Textbox(value="NAME_(STEM)_MODEL", label="Bass Stem", info="Пример вывода: Music_(Bass)_BS-Roformer-Viperx-1297", placeholder="NAME_(STEM)_MODEL")
475
  with gr.Row():
476
+ guitar_stem = gr.Textbox(value="NAME_(STEM)_MODEL", label="Guitar Stem", info="Пример вывода: Music_(Guitar)_BS-Roformer-Viperx-1297", placeholder="NAME_(STEM)_MODEL")
477
+ piano_stem = gr.Textbox(value="NAME_(STEM)_MODEL", label="Piano Stem", info="Пример вывода: Music_(Piano)_BS-Roformer-Viperx-1297", placeholder="NAME_(STEM)_MODEL")
478
 
479
  demucs_model.change(update_stems, inputs=[demucs_model], outputs=stem6)
480