karim23657 commited on
Commit
90e6952
·
verified ·
1 Parent(s): 1c7016f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -34,13 +34,8 @@ def _fn(path, solver, nfe, tau, denoising, unlimited):
34
  start = time.time()
35
  wav_onnx, new_sr = run(session, wav, sr, batch_process_chunks=False)
36
  print(f'Ran in {time.time() - start}s')
37
- # scipy.io.wavfile.write('denoiser_output.wav', new_sr, wav_onnx)
38
- audio_segment = AudioSegment(
39
- wav_onnx.tobytes(),
40
- frame_rate=new_sr,
41
- sample_width=wav_onnx.dtype.itemsize,
42
- channels=1
43
- )
44
  audio_segment.export("temp.mp3", format="mp3")
45
 
46
 
 
34
  start = time.time()
35
  wav_onnx, new_sr = run(session, wav, sr, batch_process_chunks=False)
36
  print(f'Ran in {time.time() - start}s')
37
+ scipy.io.wavfile.write('denoiser_output.wav', new_sr, wav_onnx)
38
+ audio_segment = AudioSegment.from_file('denoiser_output.wav', format="wav")
 
 
 
 
 
39
  audio_segment.export("temp.mp3", format="mp3")
40
 
41