roychao19477 commited on
Commit
18c8531
·
1 Parent(s): aeb6448
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -54,6 +54,8 @@ sdict = torch.load(ckpt, map_location=device)
54
  model.load_state_dict(sdict["generator"])
55
  model.eval()
56
 
 
 
57
  def enhance(audio, do_pcs):
58
  orig_sr, wav_np = audio
59
  # 1) resample to 16 kHz if needed
 
54
  model.load_state_dict(sdict["generator"])
55
  model.eval()
56
 
57
+
58
+ @spaces.GPU
59
  def enhance(audio, do_pcs):
60
  orig_sr, wav_np = audio
61
  # 1) resample to 16 kHz if needed