multiple reoutes for prediction

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.