fix: updated model name for PII detection feature

awslabs · Oct 18, 2023 · af5a54f · af5a54f
1 parent 575e098
commit af5a54f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/containers/lambda-containers-stack.ts b/src/containers/lambda-containers-stack.ts
@@ -14,11 +14,11 @@ export class LambdaContainersStack extends Stack {
   constructor(scope: Construct, id: string, props?: StackProps) {
     super(scope, id, props);
 
-    const ImagePIIDetection = new DockerImageFunction(
+    const DocumentPIIDetection = new DockerImageFunction(
       this,
-      'photo-pii-detection-sagemaker',
+      'document-pii-detection-sagemaker',
       {
-        functionName: 'photo-pii-detection-sagemaker',
+        functionName: 'document-pii-detection-sagemaker',
         code: DockerImageCode.fromImageAsset('src/containers/image-pii-detection',
           {
             file: 'Dockerfile.sagemaker',