add flask app to wrap prediction