Update app.py

app.py

@@ -4,17 +4,12 @@ from moviepy.editor import *
 from share_btn import community_icon_html, loading_icon_html, share_js
 
 caption = gr.Blocks.load(name="spaces/SRDdev/Image-Caption")
-audio_gen = gr.Blocks.load(name="spaces/haoheliu/audioldm-text-to-audio-generation")
+audio_gen = gr.Blocks.load(name="spaces/fffiloni/audioldm-text-to-audio-generation-clone")
 
 def infer(image_input):
     cap = caption(image_input, fn_index=0)
     sound = audio_gen(cap, 10, 2.5, 45, 3, fn_index=0)
-    
-    FILETOCONVERT = AudioFileClip(sound)
-    time.sleep(2)
-    FILETOCONVERT.write_audiofile("audio.wav")
-    FILETOCONVERT.close()
-    return cap, "audio.wav", gr.Group.update(visible=True)
+    return cap, sound[1], gr.Group.update(visible=True)
 
 title = """
     <div style="text-align: center; max-width: 700px; margin: 0 auto;">