Fix indentation

app.py

@@ -9,6 +9,7 @@ from huggingface_hub import snapshot_download
 from models import AudioDiffusion, DDPMScheduler
 from audioldm.audio.stft import TacotronSTFT
 from audioldm.variational_autoencoder import AutoencoderKL
+from pydub import AudioSegment
 
 # Automatic device detection
 if torch.cuda.is_available():
@@ -55,7 +56,7 @@ class Tango:
     def generate(self, prompt, steps = 100, guidance = 3, samples = 1, disable_progress = True):
         # Generate audio for a single prompt string
         with torch.no_grad():
-            latents = self.model.inference([prompt], self.scheduler, steps, guidance, samples, disable_progress = disable_progress, length = 20)
+            latents = self.model.inference([prompt], self.scheduler, steps, guidance, samples, disable_progress = disable_progress)
             mel = self.vae.decode_first_stage(latents)
             wave = self.vae.decode_to_waveform(mel)
         return wave
@@ -112,18 +113,30 @@ def text2audio(
     start = time.time()
     output_wave = tango.generate(prompt, steps, guidance, output_number)
 
-    output_filename_1 = "tmp1_.wav"
+    output_filename_1 = "tmp1.wav"
     wavio.write(output_filename_1, output_wave[0], rate = 16000, sampwidth = 2)
 
+    if (output_format == "mp3"):
+        AudioSegment.from_wav("tmp1.wav").export("tmp1.mp3", format = "mp3")
+        output_filename_1 = "tmp1.mp3"
+
     if (2 <= output_number):
-        output_filename_2 = "tmp2_.wav"
+        output_filename_2 = "tmp2.wav"
         wavio.write(output_filename_2, output_wave[1], rate = 16000, sampwidth = 2)
+
+        if (output_format == "mp3"):
+            AudioSegment.from_wav("tmp2.wav").export("tmp2.mp3", format = "mp3")
+            output_filename_2 = "tmp2.mp3"
     else:
         output_filename_2 = None
 
     if (output_number == 3):
-        output_filename_3 = "tmp3_.wav"
+        output_filename_3 = "tmp3.wav"
         wavio.write(output_filename_3, output_wave[2], rate = 16000, sampwidth = 2)
+
+        if (output_format == "mp3"):
+            AudioSegment.from_wav("tmp3.wav").export("tmp3.mp3", format = "mp3")
+            output_filename_3 = "tmp3.mp3"
     else:
         output_filename_3 = None