diff --git a/dan/predict/__init__.py b/dan/predict/__init__.py index c185b1f608328deddda4f973c8f716a28b58e2a2..c38e415c94469e0e0f3fc091f5151ccd2d03eda6 100644 --- a/dan/predict/__init__.py +++ b/dan/predict/__init__.py @@ -18,6 +18,7 @@ def add_predict_parser(subcommands) -> None: image_or_folder_input = parser.add_mutually_exclusive_group(required=True) image_or_folder_input.add_argument( "--image", + type=pathlib.Path, help="Path to the image to predict.", ) image_or_folder_input.add_argument(