constructor()

in source/super-resolution/super-resolution.ts [8:52]

• 41 lines of code
• 1 McCabe index (conditional complexity)

    constructor(app: cdk.App, id: string, props?: cdk.StackProps) {
        super(app, id, props);
        this.templateOptions.description = `(SO8023-sr) - AI Solution Kits - Super Resolution. 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/super-resolution-model/${modelVersion.valueAsString}/Real_ESRGAN_x2.onnx',
                'https://aws-gcr-solutions-assets.s3.cn-northwest-1.amazonaws.com.cn/ai-solution-kit/super-resolution-model/${modelVersion.valueAsString}/Real_ESRGAN_x4.onnx'
            ],
            handler: 'super_resolution_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: "SuperResolutionRESTAPI",
            stageName: 'prod',
            resourcePath: 'resolution',
            postFunction: human_seg_lambda.function,
        });
        // api.addMethod('POST', efs_lambda.function);
    }