class PreTrainedPipeline(): def __init__(self, path=""): # IMPLEMENT_THIS # Preload all the elements you are going to need at inference. # For instance your model, processors, tokenizer that might be needed. # This function is only called once, so do all the heavy processing I/O here""" raise NotImplementedError( "Please implement PreTrainedPipeline __init__ function" ) def __call__(self, inputs: str): """ Args: inputs (:obj:`str`): a string containing some text Return: A :obj:`PIL.Image` with the raw image representation as PIL. """ # IMPLEMENT_THIS raise NotImplementedError( "Please implement PreTrainedPipeline __call__ function" )