Model Inference Script Development

Hi, I am working on one of the sequential recommendation based algorithm. I am able to train model with training source code given on github page, have pasted link for the same. Next, I need help to develop inference script which can take one input sequence and give recommendations as output.
Github Link :- https://github.com/zfan20/STOSA