Pytorch Transformer Model Demo