Revert "clean up the code"

Reverts jxnl/flask-image-classifier-service#2