Apply suggestions

dan/ocr/predict/__init__.py

@@ -31,7 +31,7 @@ def add_predict_parser(subcommands) -> None:
     parser.add_argument(
         "--model",
         type=pathlib.Path,
-        help="Path to the directory containing the model, the YAML parameters file and the charset file to be used for prediction.",
+        help="Path to the directory containing the model, the YAML parameters file and the charset file to use for prediction.",
         required=True,
     )
     parser.add_argument(

dan/ocr/predict/inference.py

@@ -437,7 +437,7 @@ def run(
     Predict a single image save the output
     :param image: Path to the image to predict.
     :param image_dir: Path to the folder where the images to predict are stored.
-    :param model: Path to the directory containing the model, the YAML parameters file and the charset file to be used for prediction.
+    :param model: Path to the directory containing the model, the YAML parameters file and the charset file to use for prediction.
     :param output: Path to the output folder where the results will be saved.
     :param confidence_score: Whether to compute confidence score.
     :param attention_map: Whether to plot the attention map.

docs/usage/predict/index.md

@@ -9,7 +9,7 @@ Use the `teklia-dan predict` command to apply a trained DAN model on an image.
 | `--image`                   | Path to the image to predict. Must not be provided with `--image-dir`.                                                                | `pathlib.Path` |               |
 | `--image-dir`               | Path to the folder where the images to predict are stored. Must not be provided with `--image`.                                       | `pathlib.Path` |               |
 | `--image-extension`         | The extension of the images in the folder. Ignored if `--image-dir` is not provided.                                                  | `str`          | .jpg          |
-| `--model`                   | Path to the directory containing the model, the YAML parameters file and the charset file to be used for prediction                   | `pathlib.Path` |               |
+| `--model`                   | Path to the directory containing the model, the YAML parameters file and the charset file to use for prediction.                      | `pathlib.Path` |               |
 | `--output`                  | Path to the output folder. Results will be saved in this directory.                                                                   | `pathlib.Path` |               |
 | `--tokens`                  | Path to a yaml file containing a mapping between starting tokens and end tokens. Needed for entities.                                 | `pathlib.Path` |               |
 | `--temperature`             | Temperature scaling scalar parameter.                                                                                                 | `float`        | `1.0`         |