diff --git a/digits/model/images/classification/views.py b/digits/model/images/classification/views.py index 96b9444e6..3a8f0a4d1 100644 --- a/digits/model/images/classification/views.py +++ b/digits/model/images/classification/views.py @@ -532,7 +532,7 @@ def classify_many(): # force correct 2D shape squeezing scores for i in reversed(range(2, len(scores.shape))): if scores.shape[i] == 1: - scores = np.squeeze(scores, axis=(i,)) + scores = np.squeeze(scores, axis=(i,)) # take top 5 indices = (-scores).argsort()[:, :5]