Update app.py

app.py

@@ -85,27 +85,31 @@ 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_for_batch([prompt], steps, guidance, samples=3)
+    output_wave = tango.generate(prompt, steps, guidance)
+    output_filename_1 = "tmp.wav"
+    wavio.write(output_filename_1, output_wave[0], rate=16000, sampwidth=2)
+    # 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][0], rate=16000, sampwidth=2)
-    output_filename_2 = "tmp2.wav"
-    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[0][2], rate=16000, sampwidth=2)
+    # output_filename_1 = "tmp1.wav"
+    # 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[0][1], rate=16000, sampwidth=2)
+    # output_filename_3 = "tmp3.wav"
+    # 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")
         output_filename_1 = "tmp1.mp3"
-        AudioSegment.from_wav("tmp2.wav").export("tmp2.mp3", format = "mp3")
-        output_filename_2 = "tmp2.mp3"
-        AudioSegment.from_wav("tmp3.wav").export("tmp3.mp3", format = "mp3")
-        output_filename_3 = "tmp3.mp3"
+        # AudioSegment.from_wav("tmp1.wav").export("tmp1.mp3", format = "mp3")
+        # output_filename_1 = "tmp1.mp3"
+        # AudioSegment.from_wav("tmp2.wav").export("tmp2.mp3", format = "mp3")
+        # output_filename_2 = "tmp2.mp3"
+        # AudioSegment.from_wav("tmp3.wav").export("tmp3.mp3", format = "mp3")
+        # output_filename_3 = "tmp3.mp3"
 
-    return [output_filename_1, output_filename_2, output_filename_3]
-    # return [output_filename_1]
+    # 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/>