JakeKalstad / bee-ant-pytorch

Playing with pytorch and serving an ML prediction over HTTP.

web app to determine if an image is a bee or an ant.

Train

python3 bees.py

Predict

python3 bee_prediction.py

Web App

It's a flask app flask run