sudip1987 commited on
Commit
6917b0d
·
verified ·
1 Parent(s): 8ea47dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import scipy
3
  import gradio as gr
4
  synthesiser = pipeline("text-to-audio", "facebook/musicgen-small")
5
  def generate_music(Prompt):
6
- music = synthesiser(Prompt, forward_params={"do_sample": True, "max_new_tokens":10000})
7
  rate = music["sampling_rate"]
8
  mus = music["audio"][0].reshape(-1)
9
  return rate,mus
 
3
  import gradio as gr
4
  synthesiser = pipeline("text-to-audio", "facebook/musicgen-small")
5
  def generate_music(Prompt):
6
+ music = synthesiser(Prompt, forward_params={"do_sample": True, "max_new_tokens":100})
7
  rate = music["sampling_rate"]
8
  mus = music["audio"][0].reshape(-1)
9
  return rate,mus