I-am-vishalmaurya / pokemon-classifier

Flask application to predict Pokemon. Built with fastAI simply upload an image and the model will predict the Pokemon

Pokemon Classifier

PyTorch and FastAI implementation of pokemon classifier with over 150 different classes with the error rate of 6%.

Demo

The demo for the application is available at Pokemon Classifier

Screenshots

Installation

Create a virtual environement with python 3.7+

  python -m pip install virualenv
  python -m virtualenv venv

Activate the virtual environement

    source venv/bin/activate

Install the requirements from the application.

    pip install -r requirements.txt

Download and place the export.pkl file inside the root directory of the project. The model is avilable at export.pkl

After placing the export.pkl inside the project run the flask server with these command

    python app.py

It will start development server at http://127.0.0.1:5000