[SN-138] Modified prediction video notebook to have correct data class

[Gabefire]

Description

Please include a summary of the changes and the related issue. Please also include relevant motivation and context.

• [x] Document change (fix typo or modifying any markdown files, code comments or anything in the examples folder only)

[review-notebook-app[bot]]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB