more changes to path
Browse files- requirements.txt +2 -1
- ui/app.py +4 -2
- 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 |
-
|
|
|
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"
|
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 |
-
|
10 |
-
|
|
|
|
|
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']:
|