fffiloni commited on
Commit
a2cfdfd
·
verified ·
1 Parent(s): 64f0bfe

re-enable Riffusion model

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -289,8 +289,7 @@ def infer(image_in, chosen_model):
289
 
290
  Args:
291
  image_in: The filepath to an input image. This image is used as inspiration to generate music.
292
- chosen_model: The name of the model to use for music generation. Supported values include:
293
- "Mustango", "ACE Step".
294
 
295
  Returns:
296
  - A string containing the musical prompt generated from the image.
@@ -410,7 +409,7 @@ with gr.Blocks(css=css) as demo:
410
  #"MAGNet",
411
  "ACE Step",
412
  #"AudioLDM-2",
413
- #"Riffusion",
414
  "Mustango",
415
  #"MusicGen",
416
  #"Stable Audio Open"
 
289
 
290
  Args:
291
  image_in: The filepath to an input image. This image is used as inspiration to generate music.
292
+ chosen_model: The name of the model to use for music generation. Supported values include: "Mustango", "Riffusion", "ACE Step".
 
293
 
294
  Returns:
295
  - A string containing the musical prompt generated from the image.
 
409
  #"MAGNet",
410
  "ACE Step",
411
  #"AudioLDM-2",
412
+ "Riffusion",
413
  "Mustango",
414
  #"MusicGen",
415
  #"Stable Audio Open"