diff --git a/image_prediction/flask.py b/image_prediction/flask.py index 7ab4005..b1473da 100644 --- a/image_prediction/flask.py +++ b/image_prediction/flask.py @@ -63,7 +63,7 @@ def make_prediction_server(predict_fn: Callable): logger.info("Analysing...") predictions = predict_fn_wrapped(request.data) - if predictions: + if predictions is not None: response = jsonify(predictions) logger.info("Analysis completed.") return response