Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

multiple reoutes for prediction

Browse Source
This commit is contained in:
Matthias Bisping 2022-04-25 11:18:23 +02:00
parent c8476cb5f5
commit 6010133782
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
image_prediction/flask.py
View File

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