I want to know about how i integrate my training GAN model with GUI i train a model using pytorch that generates images. The project is that i give an input mask cover face image and model generates the image . Kindly let me know how i integrate this with GUI