INNER CODE UNIT · Python

setup

CerebriumAI/examples · 10-batching/1-litserve-batching-cpu/main.py:6

    def setup(self, device):
        print("setup")
        self.model = lambda x: x**2

    def decode_request(self, request):
        print("decode_request")
        return np.asarray(request["input"])

    def predict(self, x):
        print("predict")
        result = self.model(x)
        return result

    def encode_response(self, output):
        print("encode_response")
        return {"output": output}

View source record →