How to generate predictions from the model in the format required for the align.py file.

I want to generate outputs for a different datapoint/sequence and visualize it. How can I generate the predictions from the model in the format that the align.ipynb notebook takes? I am following the test scripts but cant seem to understand it.