ASesYusuf1 commited on
Commit
d69b111
·
verified ·
1 Parent(s): d8b9171

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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, model_file_dir, output_dir,
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
  ],