voice-isolator / example.py
huseinzolkepliscicom's picture
MB_ftlong_ws_cwlt2: weights, code, model card
35d3cc5 verified
Raw History Blame Contribute Delete
353 Bytes
"""python example.py in.wav out.wav: isolate the caller of a 16 kHz mono recording, live."""
import sys
import soundfile as sf
from vi.infer import isolate, load_model
model = load_model(".")
x, sr = sf.read(sys.argv[1], dtype="float32")
assert sr == 16000 and x.ndim == 1, "16 kHz mono"
sf.write(sys.argv[2], isolate(model, x, mode="selfl"), 16000)