ASesYusuf1 commited on
Commit
12a794c
Β·
verified Β·
1 Parent(s): b0f05b3

Update gui.py

Browse files
Files changed (1) hide show
  1. gui.py +30 -30
gui.py CHANGED
@@ -510,26 +510,26 @@ def create_interface():
510
  roformer_stem2 = gr.Audio(label="πŸ₯ Stem 2", type="filepath", interactive=False)
511
 
512
  # Auto Ensemble Sekmesi
513
- with gr.Tab("🎚️ Auto Ensemble"):
514
- with gr.Group(elem_classes="dubbing-theme"):
515
- gr.Markdown("### Ensemble Processing")
516
- with gr.Row():
517
- ensemble_audio = gr.Audio(label="🎧 Upload Audio", type="filepath", interactive=True)
518
- url_ensemble = gr.Textbox(label="πŸ”— Or Paste URL", placeholder="YouTube or audio URL", interactive=True)
519
- download_ensemble = gr.Button("⬇️ Download", variant="secondary")
520
- ensemble_exclude_stems = gr.Textbox(label="🚫 Exclude Stems", placeholder="e.g., vocals, drums (comma-separated)", interactive=True)
521
- with gr.Row():
522
- ensemble_category = gr.Dropdown(label="πŸ“š Category", choices=list(ROFORMER_MODELS.keys()), value="Instrumentals", interactive=True)
523
- ensemble_models = gr.CheckboxGroup(label="πŸ› οΈ Models", choices=list(ROFORMER_MODELS["Instrumentals"].keys()), interactive=True)
524
- with gr.Row():
525
- ensemble_seg_size = gr.Slider(32, 4000, value=256, step=32, label="πŸ“ Segment Size", interactive=True)
526
- ensemble_overlap = gr.Slider(2, 10, value=8, step=1, label="πŸ”„ Overlap", interactive=True)
527
- ensemble_use_tta = gr.Checkbox(value=False, label="πŸ” Use TTA", interactive=True)
528
- ensemble_method = gr.Dropdown(label="βš™οΈ Ensemble Method", choices=['avg_wave', 'median_wave', 'max_wave', 'min_wave', 'avg_fft', 'median_fft', 'max_fft', 'min_fft'], value='avg_wave', interactive=True)
529
- ensemble_weights = gr.Textbox(label="βš–οΈ Weights", placeholder="e.g., 1.0, 1.0 (comma-separated)", interactive=True)
530
- ensemble_button = gr.Button("πŸŽ›οΈ Run Ensemble!", variant="primary")
531
- ensemble_output = gr.Audio(label="🎢 Ensemble Result", type="filepath", interactive=False)
532
- ensemble_status = gr.Textbox(label="πŸ“’ Status", interactive=False)
533
 
534
  gr.HTML("<div class='footer'>Powered by Audio-Separator 🌟🎢 | Made with ❀️</div>")
535
 
@@ -541,16 +541,16 @@ def create_interface():
541
  inputs=[roformer_audio, roformer_model, roformer_seg_size, roformer_override_seg_size, roformer_overlap, roformer_pitch_shift, model_file_dir, output_dir, output_format, norm_threshold, amp_threshold, batch_size, roformer_exclude_stems],
542
  outputs=[roformer_stem1, roformer_stem2]
543
  )
544
- ensemble_category.change(update_ensemble_models, inputs=[ensemble_category], outputs=[ensemble_models])
545
- download_ensemble.click(fn=download_audio, inputs=[url_ensemble], outputs=[ensemble_audio])
546
- ensemble_button.click(
547
- lambda *args: auto_ensemble_process(
548
- *args[:-1],
549
- weights=[float(w.strip()) for w in args[-1].split(',')] if args[-1] else None
550
- ),
551
- inputs=[ensemble_audio, ensemble_models, ensemble_seg_size, ensemble_overlap, output_format, ensemble_use_tta, model_file_dir, output_dir, norm_threshold, amp_threshold, batch_size, ensemble_method, ensemble_exclude_stems, ensemble_weights],
552
- outputs=[ensemble_output, ensemble_status]
553
- )
554
 
555
  return app
556
 
 
510
  roformer_stem2 = gr.Audio(label="πŸ₯ Stem 2", type="filepath", interactive=False)
511
 
512
  # Auto Ensemble Sekmesi
513
+ #with gr.Tab("🎚️ Auto Ensemble"):
514
+ # with gr.Group(elem_classes="dubbing-theme"):
515
+ # gr.Markdown("### Ensemble Processing")
516
+ # with gr.Row():
517
+ # ensemble_audio = gr.Audio(label="🎧 Upload Audio", type="filepath", interactive=True)
518
+ # url_ensemble = gr.Textbox(label="πŸ”— Or Paste URL", placeholder="YouTube or audio URL", interactive=True)
519
+ # download_ensemble = gr.Button("⬇️ Download", variant="secondary")
520
+ # ensemble_exclude_stems = gr.Textbox(label="🚫 Exclude Stems", placeholder="e.g., vocals, drums (comma-separated)", interactive=True)
521
+ # with gr.Row():
522
+ # ensemble_category = gr.Dropdown(label="πŸ“š Category", choices=list(ROFORMER_MODELS.keys()), value="Instrumentals", interactive=True)
523
+ # ensemble_models = gr.CheckboxGroup(label="πŸ› οΈ Models", choices=list(ROFORMER_MODELS["Instrumentals"].keys()), interactive=True)
524
+ # with gr.Row():
525
+ # ensemble_seg_size = gr.Slider(32, 4000, value=256, step=32, label="πŸ“ Segment Size", interactive=True)
526
+ # ensemble_overlap = gr.Slider(2, 10, value=8, step=1, label="πŸ”„ Overlap", interactive=True)
527
+ # ensemble_use_tta = gr.Checkbox(value=False, label="πŸ” Use TTA", interactive=True)
528
+ # ensemble_method = gr.Dropdown(label="βš™οΈ Ensemble Method", choices=['avg_wave', 'median_wave', 'max_wave', 'min_wave', 'avg_fft', 'median_fft', 'max_fft', 'min_fft'], value='avg_wave', interactive=True)
529
+ # ensemble_weights = gr.Textbox(label="βš–οΈ Weights", placeholder="e.g., 1.0, 1.0 (comma-separated)", interactive=True)
530
+ # ensemble_button = gr.Button("πŸŽ›οΈ Run Ensemble!", variant="primary")
531
+ # ensemble_output = gr.Audio(label="🎢 Ensemble Result", type="filepath", interactive=False)
532
+ # ensemble_status = gr.Textbox(label="πŸ“’ Status", interactive=False)
533
 
534
  gr.HTML("<div class='footer'>Powered by Audio-Separator 🌟🎢 | Made with ❀️</div>")
535
 
 
541
  inputs=[roformer_audio, roformer_model, roformer_seg_size, roformer_override_seg_size, roformer_overlap, roformer_pitch_shift, model_file_dir, output_dir, output_format, norm_threshold, amp_threshold, batch_size, roformer_exclude_stems],
542
  outputs=[roformer_stem1, roformer_stem2]
543
  )
544
+ # ensemble_category.change(update_ensemble_models, inputs=[ensemble_category], outputs=[ensemble_models])
545
+ # download_ensemble.click(fn=download_audio, inputs=[url_ensemble], outputs=[ensemble_audio])
546
+ # ensemble_button.click(
547
+ # lambda *args: auto_ensemble_process(
548
+ # *args[:-1],
549
+ # weights=[float(w.strip()) for w in args[-1].split(',')] if args[-1] else None
550
+ # ),
551
+ # inputs=[ensemble_audio, ensemble_models, ensemble_seg_size, ensemble_overlap, output_format, ensemble_use_tta, model_file_dir, output_dir, norm_threshold, amp_threshold, batch_size, ensemble_method, ensemble_exclude_stems, ensemble_weights],
552
+ # outputs=[ensemble_output, ensemble_status]
553
+ # )
554
 
555
  return app
556