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.

[CLAassistant]

All committers have signed the CLA.

[LinasKo]

Hi @Lars-Codes 👋

Thank you for your contribution! Unfortunately, I can’t merge this PR due to the "Invalid Head Notebook In The Diff" issue.

I did encounter the same problems while working on #344 and have addressed them there, so I’ll be closing this PR. Thank you again for taking the time to contribute, and I hope to see more of your work in the future!