example.py

# Example usage
import soundfile as sf
from helix_model import HelixModel

# Load model from Hugging Face
model = HelixModel.from_pretrained("soundsol/helix-v0.7")

# Load audio
audio, sr = sf.read("your_audio.wav")

# Spatialize
foa = model.spatialize(
    audio,
    direction="left",
    elevation="level",
    distance="mid",
    sample_rate=sr
)

# Save FOA output
sf.write("output_spatial.wav", foa.T, 24000)