Update demos/musicgen_app.py
Browse files- 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 =
|
| 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)
|
| 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"],
|