Change train method : split adding category and training
export DJANGO_SETTINGS_MODULE in .env