Ajout d'un main + modification du code pour modele de prediction en appliquant la string documentation