Update app.py

app.py

@@ -115,13 +115,13 @@ def _do_predictions(texts, melodies, duration, progress=False, **gen_kwargs):
     outputs = outputs.detach().cpu().float()
     out_files = []
     for output in outputs:
-        with NamedTemporaryFile("wb", suffix=".mp3", delete=False) as file:
+        with NamedTemporaryFile("wb", suffix=".wav", delete=False) as file:
             audio_write(
                 file.name, output, MODEL.sample_rate, strategy="loudness",
                 loudness_headroom_db=16, loudness_compressor=True, add_suffix=False)
-            out_files.append(file.name)
+            out_files.append(pool.submit(make_waveform, file.name))
             file_cleaner.add(file.name)
-    res = [out_file.result() for out_file in out_files]
+    res = [out_file for out_file in out_files]
     for file in res:
         file_cleaner.add(file)
     print("batch finished", len(texts), time.time() - be)