Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test pretrained

This commit is contained in:
shahules786 2022-11-24 15:56:42 +05:30
parent 25139d7d3f
commit 0bca3f9949
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_inference.py
View File

@ -27,3 +27,12 @@ def test_aggregate():
data=rand, window_size=100, total_frames=1000, step_size=100 data=rand, window_size=100, total_frames=1000, step_size=100
) )
assert agg_rand.shape[-1] == 1000 assert agg_rand.shape[-1] == 1000
def test_pretrained():
from mayavoz.models import Mayamodel
model = Mayamodel.from_pretrained(
"shahules786/mayavoz-waveunet-valentini-28spk"
)
_ = model.enhance("tests/data/vctk/clean_testset_wav/p257_166.wav")