alibabasglab commited on
Commit
f98aef4
·
verified ·
1 Parent(s): 4c3cc25

Update utils/decode.py

Browse files
Files changed (1) hide show
  1. utils/decode.py +3 -1
utils/decode.py CHANGED
@@ -306,6 +306,7 @@ def _decode_one_audio_mossformergan_se_16k(model, device, inputs, norm_factor, a
306
 
307
  return outputs[:input_len].detach().cpu().numpy() # Return the output as a numpy array
308
 
 
309
  def decode_one_audio_mossformer2_se_48k(model, device, inputs, args):
310
  """Processes audio inputs through the MossFormer2 model for speech enhancement at 48kHz.
311
 
@@ -438,7 +439,8 @@ def get_mel(x, args):
438
  """
439
 
440
  return mel_spectrogram(x, args.n_fft, args.num_mels, args.sampling_rate, args.hop_size, args.win_size, args.fmin, args.fmax)
441
-
 
442
  def decode_one_audio_mossformer2_sr_48k(model, device, inputs, args):
443
  """
444
  This function decodes a single audio input using a two-stage speech super-resolution model.
 
306
 
307
  return outputs[:input_len].detach().cpu().numpy() # Return the output as a numpy array
308
 
309
+ @spaces.GPU
310
  def decode_one_audio_mossformer2_se_48k(model, device, inputs, args):
311
  """Processes audio inputs through the MossFormer2 model for speech enhancement at 48kHz.
312
 
 
439
  """
440
 
441
  return mel_spectrogram(x, args.n_fft, args.num_mels, args.sampling_rate, args.hop_size, args.win_size, args.fmin, args.fmax)
442
+
443
+ @spaces.GPU
444
  def decode_one_audio_mossformer2_sr_48k(model, device, inputs, args):
445
  """
446
  This function decodes a single audio input using a two-stage speech super-resolution model.