Gpagejr12 commited on
Commit
4a79b1a
·
verified ·
1 Parent(s): 7c39bf5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -23,16 +23,17 @@ def generate_music_tensors(descriptions, duration: int):
23
  )
24
 
25
  with st.spinner("Generating Music..."):
 
 
26
  output = model.generate(
27
  descriptions=descriptions,
28
  progress=True,
29
  return_tokens=True
30
  )
31
 
32
- st.success("Music Generation Complete!")
33
  return output
34
 
35
-
36
  def save_audio(samples: torch.Tensor):
37
  sample_rate = 30000
38
  save_path = "/audio_output"
@@ -107,4 +108,4 @@ def main():
107
 
108
 
109
  if __name__ == "__main__":
110
- main()
 
23
  )
24
 
25
  with st.spinner("Generating Music..."):
26
+ st.markdown("### Generating Music... 🎵🎶🎹")
27
+
28
  output = model.generate(
29
  descriptions=descriptions,
30
  progress=True,
31
  return_tokens=True
32
  )
33
 
34
+ st.success("Music Generation Complete! 🎉")
35
  return output
36
 
 
37
  def save_audio(samples: torch.Tensor):
38
  sample_rate = 30000
39
  save_path = "/audio_output"
 
108
 
109
 
110
  if __name__ == "__main__":
111
+ main()