Update utils/predict.py

utils/predict.py

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