download_and_train does not train the model

## Description
it takes just a few seconds to run download_and_train.py. Also, when I run predict.py it takes just a few seconds. predicted images aren't saved either. so, I don't think the model is being trained. it doesn't show any errors though. but I am not being able to train or predict. 



## Expected Behavior
download_and_train.py trains








