M4xjunior commited on
Commit
0b2a956
·
verified ·
1 Parent(s): 573718f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,11 +38,11 @@ def enhance_audio(
38
 
39
  if show_info:
40
  show_info("Denoising audio...")
41
- wav1, new_sr = denoise(dwav, sr)
42
 
43
  if show_info:
44
  show_info("Enhancing audio...")
45
- wav2, new_sr = enhance(dwav, sr, nfe=nfe, solver=solver, lambd=lambd, tau=tau)
46
 
47
  wav1 = wav1.cpu().numpy()
48
  wav2 = wav2.cpu().numpy()
 
38
 
39
  if show_info:
40
  show_info("Denoising audio...")
41
+ wav1, new_sr = denoise(dwav, sr,"cuda")
42
 
43
  if show_info:
44
  show_info("Enhancing audio...")
45
+ wav2, new_sr = enhance(dwav, sr,"cuda", nfe=nfe, solver=solver, lambd=lambd, tau=tau)
46
 
47
  wav1 = wav1.cpu().numpy()
48
  wav2 = wav2.cpu().numpy()