Problema streamlit y tensorflow

Buenas estoy tratando de correr un modelo entrenado de redes neuronales en streamlit al parecer falta un decorador llamado @keras.registrer he probado con otro formato de archivo y con el decorador y no lo puedo hacer correr

Has probado a instalarlo?

pip install --upgrade keras

Quizás también necesites estos otros:

pip install --upgrade keras-cv
pip install --upgrade keras-nlp