fffiloni commited on
Commit
78450e2
·
verified ·
1 Parent(s): a2cfdfd

re-enables AudioLDM 2 model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -289,7 +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: "Mustango", "Riffusion", "ACE Step".
293
 
294
  Returns:
295
  - A string containing the musical prompt generated from the image.
@@ -408,7 +408,7 @@ with gr.Blocks(css=css) as demo:
408
  choices = [
409
  #"MAGNet",
410
  "ACE Step",
411
- #"AudioLDM-2",
412
  "Riffusion",
413
  "Mustango",
414
  #"MusicGen",
 
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", "AudioLDM-2", "Riffusion", "ACE Step".
293
 
294
  Returns:
295
  - A string containing the musical prompt generated from the image.
 
408
  choices = [
409
  #"MAGNet",
410
  "ACE Step",
411
+ "AudioLDM-2",
412
  "Riffusion",
413
  "Mustango",
414
  #"MusicGen",