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

more changes to path

Browse files
Files changed (3) hide show
  1. requirements.txt +2 -1
  2. ui/app.py +4 -2
  3. ui/pages/rating.py +4 -2
requirements.txt CHANGED
@@ -8,4 +8,5 @@ torchmetrics==1.3.1
8
  torchvision==0.17.1
9
  streamlit==1.32.2
10
  st-star-rating==0.0.6
11
- ipython
 
 
8
  torchvision==0.17.1
9
  streamlit==1.32.2
10
  st-star-rating==0.0.6
11
+ ipython
12
+ accelerate
ui/app.py CHANGED
@@ -2,6 +2,7 @@ import streamlit as st
2
  st.set_page_config(initial_sidebar_state="collapsed")
3
  import sys
4
  sys.path.append('../')
 
5
  import generation_utilities
6
  import numpy as np
7
  import librosa
@@ -33,12 +34,13 @@ 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(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
 
2
  st.set_page_config(initial_sidebar_state="collapsed")
3
  import sys
4
  sys.path.append('../')
5
+ sys.path.append('./')
6
  import generation_utilities
7
  import numpy as np
8
  import librosa
 
34
  submit = form1.form_submit_button("Submit")
35
 
36
  if submit:
37
+ print(os.path.join(os.getcwd(),"input_songs/"))
38
+ song_list = [librosa.load(os.path.join(os.getcwd(),f"input_songs/{song}.mp3"), sr=22050)[0] for song in song_options]
39
  print(song_options,print(song_list))
40
 
41
  spectrogram, generated_song, model_name = generation_utilities.generate_songs(song_list, similarity=similarity, quality=500, merging_quality=100, device='cuda')
42
  st.session_state['song_name'] = song_options[0]
43
+ st.session_state['song_list'] = [os.path.join(os.getcwd(),f"input_songs/{song}.mp3") for song in song_options]
44
  st.session_state['song_name'] = '_'.join(song_options)
45
  st.session_state['similarity'] = similarity
46
  st.session_state['model_name'] = model_name
ui/pages/rating.py CHANGED
@@ -6,8 +6,10 @@ 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']:
 
6
  import sys
7
  import os
8
  sys.path.append('../../')
9
+ sys.path.append('../')
10
+ sys.path.append('./')
11
+ spectrograms = glob.glob(os.path.join(os.getcwd(),"ui/temp*.npy"))
12
+ generated_songs = glob.glob(os.path.join(os.getcwd(),"ui/temp*.wav"))
13
 
14
  st.markdown("# Original Song")
15
  for s in st.session_state['song_list']: