Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,10 @@ import os
|
|
6 |
import numpy as np
|
7 |
import base64
|
8 |
|
9 |
-
genres = ["Pop", "Rock", "Jazz", "Electronic", "Hip-Hop", "Classical",
|
|
|
|
|
|
|
10 |
|
11 |
@st.cache_resource()
|
12 |
def load_model():
|
@@ -19,7 +22,7 @@ def generate_music_tensors(descriptions, duration: int):
|
|
19 |
model.set_generation_params(
|
20 |
use_sampling=True,
|
21 |
top_k=250,
|
22 |
-
duration=duration
|
23 |
)
|
24 |
|
25 |
with st.spinner("Generating Music..."):
|
|
|
6 |
import numpy as np
|
7 |
import base64
|
8 |
|
9 |
+
genres = ["Pop", "Rock", "Jazz", "Electronic", "Hip-Hop", "Classical",
|
10 |
+
"Lofi", "Chillpop","Country","R&G", "Folk","Heavy Metal",
|
11 |
+
"EDM", "Soil", "Funk","Reggae", "Disco", "Punk Rock", "House",
|
12 |
+
"Techno","Indie Rock", "Grunge", "Ambient","Gospel", "Latin Music","Grime" ,"Trap", "Psychedelic Rock" ]
|
13 |
|
14 |
@st.cache_resource()
|
15 |
def load_model():
|
|
|
22 |
model.set_generation_params(
|
23 |
use_sampling=True,
|
24 |
top_k=250,
|
25 |
+
duration=duration
|
26 |
)
|
27 |
|
28 |
with st.spinner("Generating Music..."):
|