fffiloni commited on
Commit
b5357a4
β€’
1 Parent(s): caed802

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -11,7 +11,12 @@ audio_gen = gr.Blocks.load(name="spaces/fffiloni/audioldm-text-to-audio-generati
11
  def infer(image_input):
12
  cap = caption(image_input, fn_index=0)
13
  sound = audio_gen(cap, 10, 2.5, 45, 3, fn_index=0)
14
- return cap, sound[1], gr.Group.update(visible=True)
 
 
 
 
 
15
 
16
  title = """
17
  <div style="text-align: center; max-width: 700px; margin: 0 auto;">
 
11
  def infer(image_input):
12
  cap = caption(image_input, fn_index=0)
13
  sound = audio_gen(cap, 10, 2.5, 45, 3, fn_index=0)
14
+
15
+ FILETOCONVERT = AudioFileClip(sound[0])
16
+ FILETOCONVERT.write_audiofile("audio.wav")
17
+ FILETOCONVERT.close()
18
+
19
+ return cap, "audio.wav", gr.Group.update(visible=True)
20
 
21
  title = """
22
  <div style="text-align: center; max-width: 700px; margin: 0 auto;">