ai-tube-model-musicgen-1

Running on A10G

adefossez commited on Jun 9, 2023

Commit

14b5718

•

1 Parent(s): 48384f4

fix batching

Files changed (1) hide show

app_batched.py CHANGED Viewed

@@ -58,8 +58,8 @@ def predict(texts, melodies):
     for output in outputs:
         with NamedTemporaryFile("wb", suffix=".wav", delete=False) as file:
             audio_write(file.name, output, MODEL.sample_rate, strategy="loudness", add_suffix=False)
-            out_files.append([file.name])
-    return out_files
 with gr.Blocks() as demo:

     for output in outputs:
         with NamedTemporaryFile("wb", suffix=".wav", delete=False) as file:
             audio_write(file.name, output, MODEL.sample_rate, strategy="loudness", add_suffix=False)
+            out_files.append(file.name)
+    return [out_files]
 with gr.Blocks() as demo: