add images to be predicted.
add flags to load script to handle train and predict datasets
Add prediction script.