This commit is contained in:
shahules786 2022-10-06 21:08:20 +05:30
parent 2e0a6b636b
commit 69ece5b811
1 changed files with 3 additions and 1 deletions

View File

@ -139,7 +139,9 @@ class Inference:
if filename.is_file(): if filename.is_file():
raise FileExistsError(f"file {filename} already exists") raise FileExistsError(f"file {filename} already exists")
else: else:
wavfile.write(filename, rate=sr, data=waveform.detach().cpu()) wavfile.write(
filename, rate=sr, data=waveform.detach().cpu().numpy()
)
@staticmethod @staticmethod
def prepare_output( def prepare_output(