Code for prediction given a input sequence

[anirbansen2709]

The prediction code that we have is to predict on a split from the training set. Are you guys working on inference on a new sequence?
Lets say we have a model trained and we have a sequence 0 -> 1 -> 2. We would want to know what would follow (similarly for time taken)