baakaani commited on
Commit
a9c5f00
·
1 Parent(s): d22a55f

chnages to path

Browse files
Files changed (2) hide show
  1. ui/app.py +3 -3
  2. ui/pages/rating.py +3 -2
ui/app.py CHANGED
@@ -33,12 +33,12 @@ similarity = form1.slider("Similarity", 0.0, 1.0, similarity_default)
33
  submit = form1.form_submit_button("Submit")
34
 
35
  if submit:
36
- song_list = [librosa.load(f"../input_songs/{song}.mp3", sr=22050)[0] for song in song_options]
37
- print(song_options)
38
 
39
  spectrogram, generated_song, model_name = generation_utilities.generate_songs(song_list, similarity=similarity, quality=500, merging_quality=100, device='cuda')
40
  st.session_state['song_name'] = song_options[0]
41
- st.session_state['song_list'] = [f"../input_songs/{song}.mp3" for song in song_options]
42
  st.session_state['song_name'] = '_'.join(song_options)
43
  st.session_state['similarity'] = similarity
44
  st.session_state['model_name'] = model_name
 
33
  submit = form1.form_submit_button("Submit")
34
 
35
  if submit:
36
+ song_list = [librosa.load(os.path.join(os.getcwd(),f"../input_songs/{song}.mp3"), sr=22050)[0] for song in song_options]
37
+ print(song_options,print(song_list))
38
 
39
  spectrogram, generated_song, model_name = generation_utilities.generate_songs(song_list, similarity=similarity, quality=500, merging_quality=100, device='cuda')
40
  st.session_state['song_name'] = song_options[0]
41
+ st.session_state['song_list'] = [os.path.join(os.getcwd(),f"../input_songs/{song}.mp3") for song in song_options]
42
  st.session_state['song_name'] = '_'.join(song_options)
43
  st.session_state['similarity'] = similarity
44
  st.session_state['model_name'] = model_name
ui/pages/rating.py CHANGED
@@ -4,9 +4,10 @@ st.set_page_config(initial_sidebar_state="collapsed")
4
  import glob
5
  import shutil
6
  import sys
 
7
  sys.path.append('../../')
8
- spectrograms = glob.glob("temp*.npy")
9
- generated_songs = glob.glob("temp*.wav")
10
 
11
  st.markdown("# Original Song")
12
  for s in st.session_state['song_list']:
 
4
  import glob
5
  import shutil
6
  import sys
7
+ import os
8
  sys.path.append('../../')
9
+ spectrograms = glob.glob(os.path.join(os.getcwd(),"temp*.npy"))
10
+ generated_songs = glob.glob(os.path.join(os.getcwd(),"temp*.wav"))
11
 
12
  st.markdown("# Original Song")
13
  for s in st.session_state['song_list']: