Model can't be registered on MLFLOW - transformer

[hahmad2008]

Hi Facundo,
Thank you for providing an example to save the transformer model on MLFLOW.
I tried to run bert-for-classification.ipynb , the model is saved at Artifact for the current run, but it can't be registered to the model registry. do you know why?

Note: I can registry the artifact model manually from UI, but I need to do it via code.

Thanks