feat: add layout analysis

awslabs · Mar 5, 2024 · dd592a4 · dd592a4
1 parent 6083b60
commit dd592a4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/containers/lambda-containers-stack.ts b/src/containers/lambda-containers-stack.ts
@@ -14,12 +14,12 @@ export class LambdaContainersStack extends Stack {
   constructor(scope: Construct, id: string, props?: StackProps) {
     super(scope, id, props);
 
-    const DocumentPIIDetection = new DockerImageFunction(
+    const LayoutAnalysisSageMaker = new DockerImageFunction(
       this,
-      'document-pii-detection-sagemaker',
+      'layout-analysis-sagemaker',
       {
-        functionName: 'document-pii-detection-sagemaker',
-        code: DockerImageCode.fromImageAsset('src/containers/image-pii-detection',
+        functionName: 'layout-analysis-sagemaker',
+        code: DockerImageCode.fromImageAsset('src/containers/layout-analysis/model',
           {
             file: 'Dockerfile.sagemaker',
           },