diff --git a/image_prediction/flask.py b/image_prediction/flask.py index 7f76775..7ab4005 100644 --- a/image_prediction/flask.py +++ b/image_prediction/flask.py @@ -53,6 +53,7 @@ def make_prediction_server(predict_fn: Callable): return response @app.route("/predict", methods=["POST"]) + @app.route("/", methods=["POST"]) def predict(): # Tensorflow does not free RAM. Workaround: Run prediction function (which instantiates a model) in sub-process.