Spaces:
Running
on
Zero
Running
on
Zero
Update utils/decode.py
Browse files- 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.
|