Updates in the first notebook

[oliviermeslin]

The "Stage" section. Here you can indicate to all members of the project what is stage of this specific model. Let's transition it to "Production" to indicate that is our reference model, which we want to deploy

The MLFlow UI says: "Stages have been deprecated in the new Model Registry UI". I guess the only thing we can do is add a version number to the model.

The snippet model = mlflow.pyfunc.load_model(model_uri=f"models:/{model_name}/{stage}") does not work, probably because stage is deprecated.