Update audio_style_transfer/models/timedomain.py

audio_style_transfer/models/timedomain.py

@@ -10,6 +10,7 @@ import argparse
 import glob
 import os
 os.environ["CUDA_VISIBLE_DEVICES"]="-1"
+import soundfile as sf
 
 
 def chop(signal, hop_size=256, frame_size=512):
@@ -321,9 +322,9 @@ def run(content_fname,
         iterations=iterations)
 
     s = unchop(result, hop_size=hop_length, frame_size=frame_size)
-    librosa.output.write_wav(output_fname, s, sr=sr)
+    sf.write(output_fname, s, sr=sr)
     s = utils.limiter(s)
-    librosa.output.write_wav(output_fname + '.limiter.wav', s, sr=sr)
+    sf.write(output_fname+'.limiter.wav', s, sr=sr)
 
 
 def batch(content_path, style_path, output_path, model):