Update app.py

app.py

@@ -119,11 +119,13 @@ def _do_predictions(texts, melodies, duration, progress=False, **gen_kwargs):
             audio_write(
                 file.name, output, MODEL.sample_rate, strategy="loudness",
                 loudness_headroom_db=16, loudness_compressor=True, add_suffix=False)
-            out_files.append(pool.submit(make_waveform, file.name))
+            out_files.append(file.name)  # Store the filename as a string
             file_cleaner.add(file.name)
     res = [out_file for out_file in out_files]
     for file in res:
-        file_cleaner.add(file.result())  # Extract the filename from the Future object
+        if isinstance(file, Future):  # Check if it's a Future object
+            file = file.result()  # Extract the filename from the Future object
+        file_cleaner.add(file)
     print("batch finished", len(texts), time.time() - be)
     print("Tempfiles currently stored: ", len(file_cleaner.files))
     return res