**Problem:** Current implementation takes bulk dataset to train the model. If the dataset is huge, memory availability becomes a problem **Solution:** Implement batch training. Few pointers to start - - https://www.pyimagesearch.com/2018/12/24/how-to-use-keras-fit-and-fit_generator-a-hands-on-tutorial/ - https://stackoverflow.com/questions/38805375/keras-batch-training-for-multiple-large-datasets