From 158252bec6ea8a4d95406ced0907d491d68553ae Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Mon, 24 Jul 2023 11:39:49 +0000
Subject: [PATCH] Document batch prediction arg

---
 docs/usage/predict.md | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docs/usage/predict.md b/docs/usage/predict.md
index 94a241dc..294e67aa 100644
--- a/docs/usage/predict.md
+++ b/docs/usage/predict.md
@@ -23,6 +23,7 @@ Use the `teklia-dan predict` command to apply a trained DAN model on an image.
 | `--line-separators`         | List of line separators.                                                                        | `list`  | `["\n"]`      |
 | `--threshold-method`        | Method to use for attention mask thresholding. Should be in `["otsu", "simple"]`.               | `str`   | `"otsu"`      |
 | `--threshold-value `        | Threshold to use for the "simple" thresholding method.                                          | `int`   | `0`           |
+| `--batch-size `             | Size of the batches for prediction.                                                             | `int`   | `1`           |
 
 ## Examples
 
-- 
GitLab