Fabrice-TIERCELIN commited on
Commit
ab389b1
·
verified ·
1 Parent(s): 283cd1d

More explanation

Browse files
Files changed (1) hide show
  1. demos/musicgen_app.py +5 -5
demos/musicgen_app.py CHANGED
@@ -260,9 +260,9 @@ def ui_full(launch_kwargs):
260
  melody = gr.Audio(source="upload", type="numpy", label="File",
261
  interactive=True, elem_id="melody-input")
262
  with gr.Row():
263
- submit = gr.Button("Submit")
264
  # Adapted from https://github.com/rkfg/audiocraft/blob/long/app.py, MIT license.
265
- _ = gr.Button("Interrupt").click(fn=interrupt, queue=False)
266
  with gr.Row():
267
  model = gr.Radio(["facebook/musicgen-melody", "facebook/musicgen-medium", "facebook/musicgen-small",
268
  "facebook/musicgen-large",
@@ -278,9 +278,9 @@ def ui_full(launch_kwargs):
278
  with gr.Row():
279
  duration = gr.Slider(minimum=1, maximum=120, value=10, label="Duration", interactive=True)
280
  with gr.Row():
281
- topk = gr.Number(label="Top-k", value=250, interactive=True)
282
- topp = gr.Number(label="Top-p", value=0, interactive=True)
283
- temperature = gr.Number(label="Temperature", value=1.0, interactive=True)
284
  cfg_coef = gr.Number(label="Classifier Free Guidance", value=3.0, interactive=True)
285
  with gr.Column():
286
  output = gr.Video(label="Generated Music")
 
260
  melody = gr.Audio(source="upload", type="numpy", label="File",
261
  interactive=True, elem_id="melody-input")
262
  with gr.Row():
263
+ submit = gr.Button("Generate")
264
  # Adapted from https://github.com/rkfg/audiocraft/blob/long/app.py, MIT license.
265
+ _ = gr.Button("Abort all").click(fn=interrupt, queue=False)
266
  with gr.Row():
267
  model = gr.Radio(["facebook/musicgen-melody", "facebook/musicgen-medium", "facebook/musicgen-small",
268
  "facebook/musicgen-large",
 
278
  with gr.Row():
279
  duration = gr.Slider(minimum=1, maximum=120, value=10, label="Duration", interactive=True)
280
  with gr.Row():
281
+ topk = gr.Number(label="Top-k", info="Number of tokens shortlisted", value=250, interactive=True)
282
+ topp = gr.Number(label="Top-p", info="Percent of tokens shortlisted", value=0, interactive=True)
283
+ temperature = gr.Number(label="Temperature", info="lower=Always similar, higher=More creative", value=1.0, interactive=True)
284
  cfg_coef = gr.Number(label="Classifier Free Guidance", value=3.0, interactive=True)
285
  with gr.Column():
286
  output = gr.Video(label="Generated Music")