It is no doubt that doing a data science and machine learning project, starting from collecting the data, processing the data, visualizing insights about the data, and developing a machine learning model to do a predictive task is a fun thing to do.

Creating a web app is one of the solutions such that other people can make use of our machine learning model. Fortunately, it is very simple to create a web app nowadays using Streamlit library of Python

For this image classification example, we have used dataset available on Kaggle website

Overall there are three steps followed in this project:

• Creating the first Python file to load the data, build the model, and finally train the model.
• Creating the second Python file to load the model and to build the web app.
• Deploying the web app using Heroku.

Few important points related to Cat and Dog project:

• Created a website that predicts whether uploaded image is of a cat or a dog. We have used Heroku platform for deloyment.
• Used Sequential model of Keras library that allows us to create models layer-by-layer for most problems. It is eco-friendly and it is very easy to create neural network models.
• After training the model, it is store in HDF5 format (HDF5 file stands for Hierarchical Data Format 5. It is an open-source file which comes in handy to store large amount of data).
• Created a webpage using Streamlit library of Python. It creates a web interface that allows to upload image for prediction.

Link to GitHub Repository

• Deep Learning