Update app.py

app.py

@@ -85,15 +85,16 @@ tango.model.to(device_type)
     
 @spaces.GPU(duration=120)
 def gradio_generate(prompt, output_format, steps, guidance):
-    output_wave = tango.generate(prompt, steps, guidance)
+    # output_wave = tango.generate(prompt, steps, guidance)
+    output_wave = tango.generate_for_batch(prompt, steps, guidance, samples=3)
     # output_filename = f"{prompt.replace(' ', '_')}_{steps}_{guidance}"[:250] + ".wav"
 
     output_filename_1 = "tmp1.wav"
-    wavio.write(output_filename_1, output_wave[0], rate=16000, sampwidth=2)
+    wavio.write(output_filename_1, output_wave[0][0], rate=16000, sampwidth=2)
     output_filename_2 = "tmp2.wav"
-    wavio.write(output_filename_2, output_wave[1], rate=16000, sampwidth=2)
+    wavio.write(output_filename_2, output_wave[0][1], rate=16000, sampwidth=2)
     output_filename_3 = "tmp3.wav"
-    wavio.write(output_filename_3, output_wave[2], rate=16000, sampwidth=2)
+    wavio.write(output_filename_3, output_wave[0][2], rate=16000, sampwidth=2)
 
     if (output_format == "mp3"):
         AudioSegment.from_wav("tmp1.wav").export("tmp1.mp3", format = "mp3")
@@ -104,6 +105,7 @@ def gradio_generate(prompt, output_format, steps, guidance):
         output_filename_3 = "tmp3.mp3"
 
     return [output_filename_1, output_filename_2, output_filename_3]
+    # return [output_filename_1]
 
 # description_text = """
 # <p><a href="https://huggingface.co/spaces/declare-lab/tango/blob/main/app.py?duplicate=true"> <img style="margin-top: 0em; margin-bottom: 0em" src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a> For faster inference without waiting in queue, you may duplicate the space and upgrade to a GPU in the settings. <br/><br/>