Need to write inference code for generating faces and then verify the model architecture