Update inference.py

inference.py

@@ -11,7 +11,8 @@ from lib import dataset
 from lib import nets
 from lib import spec_utils
 from lib import utils
-
+import io
+from pydub import AudioSegment
 
 class Separator(object):
 
@@ -162,12 +163,17 @@ def main():
     wave = spec_utils.spectrogram_to_wave(y_spec, hop_length=args.hop_length)
     print('done')
     # sf.write('{}{}_Instruments.wav'.format(output_dir, basename), wave.T, sr)
-    sf.write('{}Instruments.wav'.format(output_dir), wave.T, sr)
-
-    print('inverse stft of vocals...', end=' ')
-    wave = spec_utils.spectrogram_to_wave(v_spec, hop_length=args.hop_length)
-    print('done')
-    sf.write('{}{}_Vocals.wav'.format(output_dir, basename), wave.T, sr)
+    #sf.write('{}Instruments.wav'.format(output_dir), wave.T, sr)
+    wav_io = io.BytesIO()
+    sf.write(wav_io, wave.T, sr,format='WAV')
+    wav_io.seek(0)
+    song=AudioSegment.from_wav(wav_io)
+    song.export('{}Instruments.ogg'.format(output_dir),format='ogg')
+    
+    #print('inverse stft of vocals...', end=' ')
+    #wave = spec_utils.spectrogram_to_wave(v_spec, hop_length=args.hop_length)
+    #print('done')
+    #sf.write('{}{}_Vocals.wav'.format(output_dir, basename), wave.T, sr)
 
     if args.output_image:
         image = spec_utils.spectrogram_to_image(y_spec)