bypass audioldm waveform saving to predictably name output files

README.md

@@ -3,6 +3,7 @@ title: AudioLDM Variations HARP plugin
 emoji: 💻
 colorFrom: blue
 colorTo: yellow
+python_version: 3.8
 sdk: gradio
 sdk_version: 4.15.0
 app_file: app.py

app.py

@@ -2,22 +2,24 @@ from pyharp import ModelCard, build_endpoint, save_and_return_filepath
 from audiotools import AudioSignal
 from audioldm import build_model, save_wave, text_to_audio
 import gradio as gr
+import soundfile as sf
 
 audioldm = build_model(model_name="audioldm-m-full")
 
 def process_fn(input_audio_path, seed, guidance_scale, num_inference_steps, num_candidates, audio_length_in_s):
-                waveform = text_to_audio(
-                    audioldm,
-                    'placeholder', 
-                    input_audio_path, 
-                    seed = int(seed),
-                    duration = audio_length_in_s, 
-                    guidance_scale = guidance_scale, 
-                    n_candidate_gen_per_text = int(num_candidates), 
-                    ddim_steps = int(num_inference_steps)
-                    )
-
-                save_wave(waveform, "./", name="output.wav")
+    waveform = text_to_audio(
+        audioldm,
+        'placeholder', 
+        input_audio_path, 
+        seed = int(seed),
+        duration = audio_length_in_s, 
+        guidance_scale = guidance_scale, 
+        n_candidate_gen_per_text = int(num_candidates), 
+        ddim_steps = int(num_inference_steps)
+        )
+    
+    sf.write("./output.wav", waveform[0, 0], samplerate=16000)
+    #save_wave(waveform, "./", name="output.wav")
 
 
 card = ModelCard(