Update utils/predict.py

utils/predict.py

@@ -62,7 +62,7 @@ def predict_batch(image_path):
     num =  random.randint(0,50)
     for images, labels in slice:
       for i,j in zip(range(num,num+3), range(9)):
-            ax = plt.subplot(1, 3, j + 1)
+            ax = plt.subplot(3, 3, j + 1)
             plt.imshow(images[i].numpy().astype("uint8"))
             output = np.argmax(slice_pred[i])