From 9e0877e150052bccaf17a5d6b5d8caa72e93b1e9 Mon Sep 17 00:00:00 2001
From: Lorenzo <lorenzobertoni9@gmail.com>
Date: Mon, 22 Mar 2021 15:42:25 +0100
Subject: [PATCH] fix printing directory

---
 monoloco/__init__.py | 2 +-
 monoloco/predict.py  | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/monoloco/__init__.py b/monoloco/__init__.py
index fdbea92..5d6032a 100644
--- a/monoloco/__init__.py
+++ b/monoloco/__init__.py
@@ -3,4 +3,4 @@
 Open implementation of MonoLoco / MonoLoco++ / MonStereo
 """
 
-__version__ = '0.5.0'
+__version__ = '0.5'
diff --git a/monoloco/predict.py b/monoloco/predict.py
index 37a09c7..c0a8fef 100644
--- a/monoloco/predict.py
+++ b/monoloco/predict.py
@@ -49,7 +49,7 @@ def download_checkpoints(args):
     dic_models = {'keypoints': pifpaf_model}
     if not os.path.exists(pifpaf_model):
         import gdown
-        LOG.info("Downloading OpenPifPaf model in %s".format(torch_dir))
+        LOG.info(f'Downloading OpenPifPaf model in {torch_dir}')
         gdown.download(OPENPIFPAF_MODEL, pifpaf_model, quiet=False)
 
     if args.mode == 'keypoints':
@@ -68,7 +68,7 @@ def download_checkpoints(args):
 
     if not os.path.exists(model):
         import gdown
-        LOG.info("Downloading model in %s".format(torch_dir))
+        LOG.info(f'Downloading model (modality: {args.mode}) in {torch_dir}')
         gdown.download(path, model, quiet=False)
     return dic_models