mashdemy commited on
Commit
d00d71c
1 Parent(s): f395d3e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
  synthesiser = pipeline("text-to-audio", "facebook/musicgen-small")
6
 
7
  def generate_music(Prompt):
8
- music = synthesiser(Prompt, forward_params={"do_sample": True, "max_new_tokens":50})
9
 
10
  # scipy.io.wavfile.write("musicgen_out2.wav", rate=music["sampling_rate"], data=music["audio"])
11
  rate = music["sampling_rate"]
 
5
  synthesiser = pipeline("text-to-audio", "facebook/musicgen-small")
6
 
7
  def generate_music(Prompt):
8
+ music = synthesiser(Prompt, forward_params={"do_sample": True, "max_new_tokens":100})
9
 
10
  # scipy.io.wavfile.write("musicgen_out2.wav", rate=music["sampling_rate"], data=music["audio"])
11
  rate = music["sampling_rate"]