From 3167612a326cd97a2e1b4ee79c323a681a9021c0 Mon Sep 17 00:00:00 2001 From: Lorenzo Date: Tue, 18 May 2021 10:45:37 +0200 Subject: [PATCH] change text --- monoloco/predict.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/monoloco/predict.py b/monoloco/predict.py index d426431..95c42b0 100644 --- a/monoloco/predict.py +++ b/monoloco/predict.py @@ -61,7 +61,8 @@ def download_checkpoints(args): pifpaf_model = args.checkpoint dic_models = {'keypoints': pifpaf_model} if not os.path.exists(pifpaf_model): - assert DOWNLOAD is not None, "install gdown to download pifpaf model, or pass it as --checkpoint" + assert DOWNLOAD is not None, \ + "pip install gdown to download a pifpaf model, or pass the model path as --checkpoint" LOG.info('Downloading OpenPifPaf model in %s', torch_dir) DOWNLOAD(OPENPIFPAF_MODEL, pifpaf_model, quiet=False) @@ -86,7 +87,8 @@ def download_checkpoints(args): dic_models[args.mode] = model if not os.path.exists(model): os.makedirs(torch_dir, exist_ok=True) - assert DOWNLOAD is not None, "pip install gdown to download monoloco model, or pass it as --model" + assert DOWNLOAD is not None, \ + "pip install gdown to download a monoloco model, or pass the model path as --model" LOG.info('Downloading model in %s', torch_dir) DOWNLOAD(path, model, quiet=False) return dic_models