fix: add default timeout during TF serving request

openfoodfacts · Dec 9, 2022 · 9aaceea · 9aaceea
1 parent a94632e
commit 9aaceea
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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()