COPIED THE WHOLE REPO FROM https://github.com/simonw/cougar-or-not
My first attempt at a machine learning API, using a pre-calculated model trained using iNaturalist data.
The resulting API is used by the @critter_vision Twitter bot, the source code for which can be found at https://github.com/natbat/CritterVision
The model is usa-inaturalist-cats.pth - an 83MB file.
The notebook inaturalist-cats.ipynb shows how I trained the model, using fastai.
cougar.py is a very tiny Starlette API server which simply accepts file image uploads and runs them against the pre-calculated model.
It also accepts a URL to an image, e.g. https://cougar-or-not.now.sh/classify-url?url=https://upload.wikimedia.org/wikipedia/commons/9/9a/Oregon_Cougar_ODFW.JPG
The Dockerfile means the entire thing can be deployed to Zeit Now or any other container hosting service.
