adding pytorch model

[gaetanbrison]

The goal of this commit is to add an example of application of classification of images using as input mnist dataset and pytorch lightning model through mlflow.