machine learning model using django