created a model on colab but cannot load for inference

[sedici16]

I've successfully created a model on Google Colab, but I'm encountering issues when trying to load it for inference. Could you provide detailed guidance or documentation on the correct procedures for loading and utilizing the model

https://colab.research.google.com/drive/1Q8S5OIC3PNmayZWCI1voLxLKZYliVEYF?usp=sharing

is there a procedure to extract the weights from the training and use it the standard HF transformer?