lorneluo commited on
Commit
8f14b67
1 Parent(s): 74a8d2d

minor changes

Browse files
Files changed (3) hide show
  1. args.py +2 -2
  2. inst.py +4 -3
  3. install_packages.sh +0 -5
args.py CHANGED
@@ -300,7 +300,7 @@ uvr_mdx_net_voc_ft_process_data = {
300
  'audio_file': None,
301
  'set_progress_bar': lambda step, inference_iterations=0: print(
302
  f"iteration {inference_iterations} of step #{step}"),
303
- 'write_to_console': lambda progress_text, base_text='base_text': print(
304
  f"{progress_text} {base_text}"),
305
  'process_iteration': lambda iteration: iteration + 1,
306
  'cached_source_callback': cached_source_callback,
@@ -424,7 +424,7 @@ htdemucs_ft_process_data = {
424
  'audio_file': None,
425
  'set_progress_bar': lambda step, inference_iterations=0: print(
426
  f"iteration {inference_iterations} of step #{step}"),
427
- 'write_to_console': lambda progress_text, base_text='base_text': print(
428
  f"{progress_text} {base_text}"),
429
  'process_iteration': lambda iteration: iteration + 1,
430
  'cached_source_callback': cached_source_callback,
 
300
  'audio_file': None,
301
  'set_progress_bar': lambda step, inference_iterations=0: print(
302
  f"iteration {inference_iterations} of step #{step}"),
303
+ 'write_to_console': lambda progress_text, base_text='': print(
304
  f"{progress_text} {base_text}"),
305
  'process_iteration': lambda iteration: iteration + 1,
306
  'cached_source_callback': cached_source_callback,
 
424
  'audio_file': None,
425
  'set_progress_bar': lambda step, inference_iterations=0: print(
426
  f"iteration {inference_iterations} of step #{step}"),
427
+ 'write_to_console': lambda progress_text, base_text='': print(
428
  f"{progress_text} {base_text}"),
429
  'process_iteration': lambda iteration: iteration + 1,
430
  'cached_source_callback': cached_source_callback,
inst.py CHANGED
@@ -3,7 +3,7 @@ import shutil
3
  import sys
4
  from datetime import datetime
5
  from pathlib import Path
6
-
7
 
8
  from args import mdx23c_8kfft_instvoc_hq_process_data, htdemucs_ft_process_data, uvr_mdx_net_voc_ft_process_data
9
  from download import download_model, get_model_file
@@ -66,11 +66,12 @@ def run_ensemble_models(audio_path, export_path, format=WAV, clean=True):
66
  ensemble(vocals_export_paths, vocals_final_path)
67
  ensemble(instrumental_export_paths, instrumental_final_path)
68
 
69
- if clean:
70
- shutil.rmtree(temp_export_path)
71
  print(f'instrumental_final_path', instrumental_final_path)
72
  print(f'vocals_final_path', vocals_final_path)
73
  print(f'Finished in {datetime.now() - start}')
 
 
 
74
  return instrumental_final_path, vocals_final_path
75
 
76
 
 
3
  import sys
4
  from datetime import datetime
5
  from pathlib import Path
6
+ from time import sleep
7
 
8
  from args import mdx23c_8kfft_instvoc_hq_process_data, htdemucs_ft_process_data, uvr_mdx_net_voc_ft_process_data
9
  from download import download_model, get_model_file
 
66
  ensemble(vocals_export_paths, vocals_final_path)
67
  ensemble(instrumental_export_paths, instrumental_final_path)
68
 
 
 
69
  print(f'instrumental_final_path', instrumental_final_path)
70
  print(f'vocals_final_path', vocals_final_path)
71
  print(f'Finished in {datetime.now() - start}')
72
+ if clean:
73
+ sleep(10)
74
+ shutil.rmtree(temp_export_path)
75
  return instrumental_final_path, vocals_final_path
76
 
77
 
install_packages.sh DELETED
@@ -1,5 +0,0 @@
1
- #!/bin/bash
2
-
3
- while read package; do
4
- pip install "$package"
5
- done < requirements.txt