pipeline {
    docker {
            image 'ksero/ium:pytorch'
        }
    stages {
        stage('Prediction') {
            steps {
                sh 'ls -la'
                sh "python ./predictMlflow.py"
            }
        }
    }
}