From 5e09d0424cda13e2fec394abb3bac991e041c7e2 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 19 Dec 2021 17:08:15 +0100
Subject: [PATCH] Update

---
 classifier.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/classifier.py b/classifier.py
index f3b1514cc..07408ebda 100644
--- a/classifier.py
+++ b/classifier.py
@@ -245,7 +245,7 @@ def imshow(img, labels=None, pred=None, names=None, nmax=64, verbose=False, f=Pa
         ax[i].imshow(blocks[i].squeeze().permute((1, 2, 0)))  # cmap='gray'
         ax[i].axis('off')
         if labels is not None:
-            s = names[labels[i]] + f'—{names[pred[i]]}' if pred is not None else ''
+            s = names[labels[i]] + (f'—{names[pred[i]]}' if pred is not None else '')
             ax[i].set_title(s)
 
     plt.savefig(f, dpi=300, bbox_inches='tight')