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 in range(9): #zip(range(num,num+3), range(9)):
+      for i in range(6):
             ax = plt.subplot(3, 3, i + 1)
             plt.imshow(images[i].numpy().astype("uint8"))
             output = np.argmax(slice_pred[i])