in codes/lambda/api-hosting-predictor/src/handler.py [0:0]
def handle(event, context):
logger.info('handler handle: event-> {}'.format(event))
message_id = str(uuid.uuid4())
if 'sentence' in event:
sm_client = load_sm_client()
prediction = predict(sm_client, _endpoint_name, event)
if prediction is not None:
return create_success_response(message_id, prediction)
else:
return create_error_response(message_id, 'no response from sagemaker')
else:
return create_error_response(message_id, 'wrong request format')