vpavlenko commited on
Commit
62ea0b2
·
1 Parent(s): 256bf62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -51,7 +51,7 @@ HEADER = """
51
 
52
  CACHE_EXAMPLES = os.getenv('CACHE_EXAMPLES', '1') == '1'
53
 
54
- BASIC_PITCH_DIR = "/app/basic-pitch"
55
 
56
 
57
  def compress_files(demucs_output_path, dissector_file, original_audio):
@@ -87,9 +87,9 @@ def compress_files(demucs_output_path, dissector_file, original_audio):
87
  with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
88
  for mp3 in mp3_files:
89
  zipf.write(os.path.join(demucs_output_path, mp3), arcname=mp3)
90
- zipf.write('/app/basic-pitch/bass_basic_pitch.mid', arcname='bass_basic_pitch.mid')
91
- zipf.write('/app/basic-pitch/vocals_basic_pitch.mid', arcname='vocals_basic_pitch.mid')
92
- zipf.write('/app/basic-pitch/other_basic_pitch.mid', arcname='other_basic_pitch.mid')
93
  zipf.write(dissector_file, arcname='dissector.json') # Fixed name
94
  zipf.write(original_audio, arcname='mixdown.mp3') # Added original audio
95
 
 
51
 
52
  CACHE_EXAMPLES = os.getenv('CACHE_EXAMPLES', '1') == '1'
53
 
54
+ BASIC_PITCH_DIR = "basic-pitch"
55
 
56
 
57
  def compress_files(demucs_output_path, dissector_file, original_audio):
 
87
  with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
88
  for mp3 in mp3_files:
89
  zipf.write(os.path.join(demucs_output_path, mp3), arcname=mp3)
90
+ zipf.write(f'{BASIC_PITCH_DIR}/bass_basic_pitch.mid', arcname='bass_basic_pitch.mid')
91
+ zipf.write(f'{BASIC_PITCH_DIR}/vocals_basic_pitch.mid', arcname='vocals_basic_pitch.mid')
92
+ zipf.write(f'{BASIC_PITCH_DIR}/other_basic_pitch.mid', arcname='other_basic_pitch.mid')
93
  zipf.write(dissector_file, arcname='dissector.json') # Fixed name
94
  zipf.write(original_audio, arcname='mixdown.mp3') # Added original audio
95