Upload musicgen_app.py
Browse files- demos/musicgen_app.py +17 -6
demos/musicgen_app.py
CHANGED
|
@@ -317,12 +317,23 @@ def ui_full(launch_kwargs):
|
|
| 317 |
# Adapted from https://github.com/rkfg/audiocraft/blob/long/app.py, MIT license.
|
| 318 |
_ = gr.Button("Interrupt").click(fn=interrupt, queue=False)
|
| 319 |
with gr.Row():
|
| 320 |
-
|
| 321 |
-
|
| 322 |
-
|
| 323 |
-
|
| 324 |
-
|
| 325 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 326 |
model_path = gr.Text(label="Model Path (custom models)")
|
| 327 |
with gr.Row():
|
| 328 |
decoder = gr.Radio(
|
|
|
|
| 317 |
# Adapted from https://github.com/rkfg/audiocraft/blob/long/app.py, MIT license.
|
| 318 |
_ = gr.Button("Interrupt").click(fn=interrupt, queue=False)
|
| 319 |
with gr.Row():
|
| 320 |
+
model = gr.Radio(
|
| 321 |
+
[
|
| 322 |
+
"facebook/musicgen-melody",
|
| 323 |
+
"facebook/musicgen-medium",
|
| 324 |
+
"facebook/musicgen-small",
|
| 325 |
+
"facebook/musicgen-large",
|
| 326 |
+
"facebook/musicgen-melody-large",
|
| 327 |
+
"facebook/musicgen-stereo-small",
|
| 328 |
+
"facebook/musicgen-stereo-medium",
|
| 329 |
+
"facebook/musicgen-stereo-melody",
|
| 330 |
+
"facebook/musicgen-stereo-large",
|
| 331 |
+
"facebook/musicgen-stereo-melody-large",
|
| 332 |
+
],
|
| 333 |
+
label="Model",
|
| 334 |
+
value="facebook/musicgen-stereo-melody",
|
| 335 |
+
interactive=True,
|
| 336 |
+
)
|
| 337 |
model_path = gr.Text(label="Model Path (custom models)")
|
| 338 |
with gr.Row():
|
| 339 |
decoder = gr.Radio(
|