preload.py

"""
`preload.py` is run when building the docker container,
and can be used to download any assets like pretrained models.
"""
from sample_demo.model import Model

def preload():
    # this downloads the model, such that it is stored on disk, and does not need to be loaded at runtime
    model = Model()

if __name__=="__main__":
    preload()