model.py

class Model:
    def __init__(self):
        self.model = None
        self.tokenizer = None
        self.gen_config = None
        self.INPUT_FORMAT = None
    
    def generate(self, inputs:str) -> str:
        outputs = inputs
        return outputs