constructor()

in source/human-seg/human-seg.ts [8:51]


    constructor(app: cdk.App, id: string, props?: cdk.StackProps) {
        super(app, id, props);
        this.templateOptions.description = `(SO8023-seg) - AI Solution Kits - Human Image Segmentation. Template version v1.0.0`;

        const modelVersion = new cdk.CfnParameter(this, "modelVersion", {
            default: 'v1.0.0',
            type: 'String',
            description: 'Pre-trained model version, this parameter works only for testing, please do NOT change the default value.'
        });

        const human_seg_lambda = new EfsModelLambda(this, 'human_seg_lambda', {
            mountPath: '/mnt/lambda',
            objects: [
                `https://aws-gcr-solutions-assets.s3.cn-northwest-1.amazonaws.com.cn/ai-solution-kit/human-seg-model/${modelVersion.valueAsString}/humanseg_720.onnx`,
            ],
            handler: 'human_seg_app.lambda_handler',
            code: lambda.Code.fromAsset(path.join(__dirname, './src'), {
                bundling: {
                    image: lambda.Runtime.PYTHON_3_8.bundlingImage,
                    command: [
                        'bash', '-c', [
                            `cp -r /asset-input/* /asset-output/`,
                        ].join(' && ')
                    ],
                }
            }),
            runtime: lambda.Runtime.PYTHON_3_8,
            layerPath: path.join(__dirname, '/src'),
            timeout: cdk.Duration.minutes(1),
            memorySize: 10240,
            environment: {
                PYTHONPATH: '/var/runtime:/opt:/opt/python/lib/python3.8/site-packages:python:python/lib/python3.8/site-packages',
                ACCESS_POINT: '/mnt/lambda',
            }
        });

        const api = new AiKitsRestApi(this, "aikits", {
            apiName: "HumanSegRESTAPI",
            stageName: 'prod',
            resourcePath: 'seg',
            postFunction: human_seg_lambda.function,
        });
        // api.addMethod('POST', efs_lambda.function);
    }