A Flask hosted model to differentiate between cats and dogs using a trained ML model
Using the pytorch library to train a Convoluted Neural Network (CNN) model to distingusih between cats and dogs. Users can pass in images into the model from a web interface and the model outputs whether the image is that of a cat or dog.
Prerequistes:
- Python 3.x
Steps to use:
- Clone the repository
- Install all the dependencies needed - Pytorch, Flask, Pillow and Pickle
- Run the code using the terminal command
Flash Runfrom the same directory as App.py - Use the web interface to input an image into the model