Gpagejr12 commited on
Commit
b4432a0
·
verified ·
1 Parent(s): a294bf3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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", "Lofi", "Chillpop"]
 
 
 
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 * 60
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..."):