Skip to content

Latest commit

 

History

History
54 lines (28 loc) · 1.48 KB

README.md

File metadata and controls

54 lines (28 loc) · 1.48 KB

Image Classification

FastApi api that predicts if it's a cat or a dog using fastAI image classification.

demo1

  • Deployed site: CLICK HERE as I'm using a free tie to deploy this site you may ecounter limitations or the app could not be working follow the instructions bellow to run locally.

Requirements

  • Python3

Backend

  • Create a virtual enviroment running in yuor terminal:

python3 -m venv env

  • Actinvate your virtual enviroment running:

On mac: source env/bin/activate

On windows: .\env\Scripts\activate

  • Install dependecies:

pip install -r requirements.txt

  • Create image classification running in your terminal:

python3 download.py

It will create a folder classification with pictures of dogs and cats in your root. attention: You should verify if each folder contains really cats and dogs also this pretrained model tends to be biased toward cats consired add more dogs data

  • Train your model running in your terminal:

python3 train.py

It will create a file called dog_or_park.pkl in your root.

  • Run your server:

uvicorn app:app --reload

Your server will be running at http://127.0.0.1:8000

Frontend