FastApi api that predicts if it's a cat or a dog using fastAI image classification.
- 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.
- Python3
- 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
- Link to frontend repo: CLICK HERE