Update train-yolo-nas-on-custom-dataset.ipynb

[Lars-Codes]

Description

Under Single Image Interference,

This code: list(model.predict(image, conf=0.35))[0] Should become: model.predict(image, conf=0.35)

Please delete options that are not relevant.

• [ ] This change requires a documentation update

How has this change been tested, please provide a testcase or example of how you tested the change?

My code on this notebook only worked when I made this change. Otherwise it was throwing errors.