Method decode_request

tests/unit/test_lit_server.py:152–154

(self, request: Request)

        self.device = device

    def decode_request(self, request: Request):
        content = request["input"]
        return torch.tensor([content], device=self.device)

    def predict(self, x):
        return self.model(x[None, :])