fffiloni commited on
Commit
fd37796
1 Parent(s): 5e703c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -253,7 +253,7 @@ def infer(image_in, chosen_model, api_status):
253
  music_o = get_magnet(musical_prompt)
254
  elif chosen_model == "AudioLDM-2" :
255
  gr.Info("Now calling AudioLDM-2 for music...")
256
- music_o = get_magnet(musical_prompt)
257
  elif chosen_model == "Riffusion" :
258
  gr.Info("Now calling Riffusion for music...")
259
  music_o = get_riffusion(musical_prompt)
@@ -274,7 +274,7 @@ def retry(chosen_model, caption):
274
  music_o = get_magnet(musical_prompt)
275
  elif chosen_model == "AudioLDM-2" :
276
  gr.Info("Now calling AudioLDM-2 for music...")
277
- music_o = get_magnet(musical_prompt)
278
  elif chosen_model == "Riffusion" :
279
  gr.Info("Now calling Riffusion for music...")
280
  music_o = get_riffusion(musical_prompt)
 
253
  music_o = get_magnet(musical_prompt)
254
  elif chosen_model == "AudioLDM-2" :
255
  gr.Info("Now calling AudioLDM-2 for music...")
256
+ music_o = get_audioldm(musical_prompt)
257
  elif chosen_model == "Riffusion" :
258
  gr.Info("Now calling Riffusion for music...")
259
  music_o = get_riffusion(musical_prompt)
 
274
  music_o = get_magnet(musical_prompt)
275
  elif chosen_model == "AudioLDM-2" :
276
  gr.Info("Now calling AudioLDM-2 for music...")
277
+ music_o = get_audioldm(musical_prompt)
278
  elif chosen_model == "Riffusion" :
279
  gr.Info("Now calling Riffusion for music...")
280
  music_o = get_riffusion(musical_prompt)