comidan commited on
Commit
77d168a
·
verified ·
1 Parent(s): 7918e8e

Update demos/musicgen_app.py

Browse files
Files changed (1) hide show
  1. demos/musicgen_app.py +6 -6
demos/musicgen_app.py CHANGED
@@ -262,12 +262,12 @@ def ui_full(launch_kwargs):
262
  # Adapted from https://github.com/rkfg/audiocraft/blob/long/app.py, MIT license.
263
  _ = gr.Button("Interrupt").click(fn=interrupt, queue=False)
264
  with gr.Row():
265
- model = "facebook/" + gr.Radio(["musicgen-melody", "musicgen-medium", "musicgen-small",
266
- "musicgen-large", "musicgen-melody-large",
267
- "musicgen-stereo-small", "musicgen-stereo-medium",
268
- "musicgen-stereo-melody", "musicgen-stereo-large",
269
- "musicgen-stereo-melody-large"],
270
- label="Model", value="musicgen-stereo-melody", interactive=True).value
271
  model_path = gr.Text(label="Model Path (custom models)")
272
  with gr.Row():
273
  decoder = gr.Radio(["Default", "MultiBand_Diffusion"],
 
262
  # Adapted from https://github.com/rkfg/audiocraft/blob/long/app.py, MIT license.
263
  _ = gr.Button("Interrupt").click(fn=interrupt, queue=False)
264
  with gr.Row():
265
+ model = gr.Radio(["facebook/musicgen-melody", "facebook/musicgen-medium", "facebook/musicgen-small",
266
+ "facebook/musicgen-large", "facebook/musicgen-melody-large",
267
+ "facebook/musicgen-stereo-small", "facebook/musicgen-stereo-medium",
268
+ "facebook/musicgen-stereo-melody", "facebook/musicgen-stereo-large",
269
+ "facebook/musicgen-stereo-melody-large"],
270
+ label="Model", value="facebook/musicgen-stereo-melody", interactive=True)
271
  model_path = gr.Text(label="Model Path (custom models)")
272
  with gr.Row():
273
  decoder = gr.Radio(["Default", "MultiBand_Diffusion"],