import torch def load_model(): model = torch.jit.load("user_0/scripted_model.pt") model.eval() return model def predict(model, inputs): inputs = [torch.tensor(input) for input in inputs] with torch.no_grad(): output = model(*inputs) return output.detach().cpu().numpy().tolist()