Slackbot
06/08/2023, 4:14 AMChaoyu
06/08/2023, 4:20 AMChaoyu
06/08/2023, 4:20 AMChaoyu
06/08/2023, 4:21 AMPaul Roopson Pradeep
06/08/2023, 4:21 AMdef my_python_model(input_list: List[int]) -> List[int]:
return np.square(np.array(input_list))
# `save_model` saves a given python object or function
saved_model = bentoml.picklable_model.save_model(
'themodel',
my_python_model,
signatures={"__call__": {"batchable": True}}
)
print(f"Model saved: {saved_model}")
Paul Roopson Pradeep
06/08/2023, 5:48 AM