update

app.py

@@ -13,6 +13,7 @@ import numpy as np
 import json
 from datetime import datetime
 import gc
+import soundfile as sf
 
 from resonate.eval_utils import generate_fm, setup_eval_logging
 from resonate.model.flow_matching import FlowMatching
@@ -187,7 +188,7 @@ def generate_audio_gradio(
         filename = f"{safe_prompt}_{current_time_string}_{i}.flac"
         save_path = OUTPUT_DIR / filename
         # torchaudio.save(str(save_path), audio, seq_cfg.sampling_rate)
-        torchaudio.save(str(save_path), audio, seq_cfg.sampling_rate, backend="soundfile")
+        sf.write(str(save_path), audio.squeeze().cpu().numpy(), seq_cfg.sampling_rate)
         log.info(f"Audio saved to {save_path}")
         save_paths.append(str(save_path))
 

requirements.txt

@@ -27,7 +27,6 @@ python-dotenv
 transformers
 debugpy
 soundfile
-torchcodec
 
 --extra-index-url https://download.pytorch.org/whl/nightly/cu128
 torchcodec