Issue in running the Yolo NAS notebook

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

Originally posted by @SkalskiP in https://github.com/roboflow/notebooks/issues/172#issuecomment-1754866035

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

Start by opening the linked train-yolo-nas-on-custom-dataset.ipynb notebook in Colab and running the cell containing cv2.imread and model.predict. Inspect the prediction API at the failing call and confirm the notebook's usage matches the returned object. Done means the notebook runs that cell without the TypeError and produces the expected prediction result.