End-to-End-Image-Classifcation I have added a GUI to try the model if you are interested. I have also added images for prediction purposes.