def METHOD_NAME(self, audio, f0):
    with torch.no_grad():
        mel = self.stft.get_mel(audio)
        enhanced_audio = self.model(mel, f0[:, : mel.size(-1)])
        return enhanced_audio, self.h.sampling_rate