Dog and Cat Breed Predictor

Submit a .jpg, .jpeg or .png image to get a prediction.


This model was trained using the data from the Dockship 'Animal Breed Classification AI Challenge'
The data is limited in scope and so the model is only able to make predictions on the following breeds:

Note: The confidence is the predicted probability of the estimation being correct, assuming that the breed is one of the 37 listed. As is becoming common knowledge with the mainstream adoption of ChatGPT, AI can be confidently wrong. This model is no different. The predictions can be incorrect and the model often gets confused between similar-looking breeds. In testing, the model predicted the correct breed 90% of the time.

Check out my:

  • github repo for the notebook used to train and test the model.
  • github repo used to deploy the app.