From af5a54fcbb8444d1cd50ac7b4b777690722312e3 Mon Sep 17 00:00:00 2001
From: Mike Yan <yiyanz@amazon.com>
Date: Wed, 18 Oct 2023 09:50:16 +0800
Subject: [PATCH] fix: updated model name for PII detection feature

---
 src/containers/lambda-containers-stack.ts | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/containers/lambda-containers-stack.ts b/src/containers/lambda-containers-stack.ts
index f4b52756..229e518d 100644
--- 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',