Add parameters to use or not use mlflow

[adimo20]

ML FLow as a paramter

When you don't want to use mlflow, the training and and other scripts break because the can't conntect to mlflow.
Task:

• Add paramter that makes it possible to not use mlflow