How to convert to waveform from the generated stft

Thanks for your code.
I wonder how to convert from enhanced stft generated from the generator to the waveform so that I can hear the enhanced audio
I know that I need to use torch.istft, but since you conduct power compression to the output of torch.stft, I don't know how to change the enhanced output before applying torch.istft.
I hope you can help me!
Thank you!