Question regarding how to extract protein representation from your pretrain model

[Tizzzzy]

Hi author,
Huge fan of your work. I am currently trying to apply your code to a downstream task. Specifically, I am working with a protein PDB file and aim to extract the latent representation of the protein using your pretrained model. I was wondering if this is possible.
If it is, could you kindly show me which script I should run and which line of code store the representation of the protein?
Thank you so much for your time