diff --git a/robotoff/prediction/category/neural/category_classifier.py b/robotoff/prediction/category/neural/category_classifier.py index 08fdb9827c..9a06d0785f 100644 --- a/robotoff/prediction/category/neural/category_classifier.py +++ b/robotoff/prediction/category/neural/category_classifier.py @@ -84,7 +84,9 @@ def predict( } r = http_session.post( - f"{settings.TF_SERVING_BASE_URL}/category-classifier:predict", json=data + f"{settings.TF_SERVING_BASE_URL}/category-classifier:predict", + json=data, + timeout=(3.0, 10.0), ) r.raise_for_status() response = r.json()