fixed prediction visualization (demos) for any number of frames.

[farhanm612]

Fixed the use of optional function parameter, to add the ability to generate visualizations for videos with any number of frames.