Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -553,7 +553,7 @@ def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, p
|
|
553 |
if os.path.exists(output_dir):
|
554 |
shutil.rmtree(output_dir)
|
555 |
os.makedirs(output_dir, exist_ok=True)
|
556 |
-
base_name = os.path.splitext(os.path.basename(audio))[0]
|
557 |
for category, models in ROFORMER_MODELS.items():
|
558 |
if model_key in models:
|
559 |
model = models[model_key]
|
@@ -691,7 +691,7 @@ def auto_ensemble_process(audio, model_keys, state, seg_size=64, overlap=0.1, ou
|
|
691 |
logger.info("New audio detected, resetting ensemble state.")
|
692 |
|
693 |
use_tta = use_tta == "True"
|
694 |
-
base_name = os.path.splitext(os.path.basename(audio))[0]
|
695 |
logger.info(f"Ensemble for {base_name} with {model_keys} on {device}")
|
696 |
|
697 |
permanent_output_dir = os.path.join(output_dir, "permanent_stems")
|
@@ -972,7 +972,7 @@ def create_interface():
|
|
972 |
fn=auto_ensemble_process,
|
973 |
inputs=[
|
974 |
ensemble_audio, ensemble_models, ensemble_state, ensemble_seg_size, ensemble_overlap,
|
975 |
-
output_format, ensemble_use_tta,
|
976 |
norm_threshold, amp_threshold, batch_size, ensemble_method,
|
977 |
ensemble_exclude_stems, ensemble_weights
|
978 |
],
|
|
|
553 |
if os.path.exists(output_dir):
|
554 |
shutil.rmtree(output_dir)
|
555 |
os.makedirs(output_dir, exist_ok=True)
|
556 |
+
base_name = os.path.splitext(os.path.basename(audio))[0].replace(' ', '_') # Boşlukları alt çizgi ile değiştir
|
557 |
for category, models in ROFORMER_MODELS.items():
|
558 |
if model_key in models:
|
559 |
model = models[model_key]
|
|
|
691 |
logger.info("New audio detected, resetting ensemble state.")
|
692 |
|
693 |
use_tta = use_tta == "True"
|
694 |
+
base_name = os.path.splitext(os.path.basename(audio))[0].replace(' ', '_') # Boşlukları alt çizgi ile değiştir
|
695 |
logger.info(f"Ensemble for {base_name} with {model_keys} on {device}")
|
696 |
|
697 |
permanent_output_dir = os.path.join(output_dir, "permanent_stems")
|
|
|
972 |
fn=auto_ensemble_process,
|
973 |
inputs=[
|
974 |
ensemble_audio, ensemble_models, ensemble_state, ensemble_seg_size, ensemble_overlap,
|
975 |
+
output_format, ensemble_use_tta, model_dir, output_dir,
|
976 |
norm_threshold, amp_threshold, batch_size, ensemble_method,
|
977 |
ensemble_exclude_stems, ensemble_weights
|
978 |
],
|