1. 必须包含 execute 函数:def execute(input_data, params=None):
2. 函数接收 input_data(字典)和 params(字典,可选)参数
3. 函数返回处理结果(字典)
4. 示例:
def execute(input_data, params=None):
a = input_data.get('a', 0)
b = input_data.get('b', 0)
return {"result": a + b}
1. 必须包含 execute 函数:def execute(input_data, params=None):
2. 函数接收 input_data(字典)和 params(字典,可选)参数
3. 函数返回处理结果(字典)
4. 示例(PyTorch模型):
import torch
# 加载模型
model = torch.load('model/model.pth')
model.eval()
def execute(input_data, params=None):
data = torch.tensor(input_data['data'])
with torch.no_grad():
output = model(data)
return {"result": output.numpy().tolist()}