peterbacalso/african_wildlife_classifier

API for identifying buffalos, elephants, rhinos, or zebras in African wildlife # African Wildlife Minimal machine learning API, inspired by fastai MOOC and cougar-or-not api `app/main.py` is a small API built with fastapi. It takes an image url or file and runs it through a pretrained model for classification between buffalo, elephant, rhino, or zebra. `export.pkl` is a resnet34 model, pretrained with the fastai framework using kaggle African Wildlife dataset. See `african_wildlife.ipynb` for training details.