Issue in running the Yolo NAS notebook

[Darrshan-Sankar]

          Thanks @arrafi-musabbir 🙏🏻 I just updated our YOLO-NAS notebook.

Originally posted by @SkalskiP in #172 (comment)

I wanted to reopen this issue, as i get error for the following code:

import cv2

image = cv2.imread(SOURCE_IMAGE_PATH)
result = list(model.predict(image, conf=0.35))[0]

I get the following error:

TypeError                                 Traceback (most recent call last)
[<ipython-input-8-71f83f1aa3bd>](https://localhost:8080/#) in <cell line: 4>()
      2 
      3 image = cv2.imread(SOURCE_IMAGE_PATH)
----> 4 result = list(model.predict(image, conf=0.35))[0]

TypeError: 'ImageDetectionPrediction' object is not iterable

I ran the following notebook: https://colab.research.google.com/github/roboflow-ai/notebooks/blob/main/notebooks/train-yolo-nas-on-custom-dataset.ipynb in colab