Merge pull request #150 from awslabs/icyxu/image-pii-detection

fix(be): Update Identifier in PII Detection
awslabs · Oct 25, 2023 · ad02ea6 · ad02ea6
2 parents d373ce1 + b885a6a
commit ad02ea6
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/containers/image-pii-detection/parsers/image_parser.py b/src/containers/image-pii-detection/parsers/image_parser.py
@@ -60,16 +60,16 @@ def parse_file(self, file_path):
 
         if contain_face:
             if contain_cnid:
-                file_content.append('ChineseID')
+                file_content.append('CHINESE_ID_IMAGE')
             elif contain_car_license:
-                file_content.append('CarLicense')
+                file_content.append('CAR_LICENSE_IMAGE')
             else:
-                file_content.append('Face')
+                file_content.append('FACE_IMAGE')
         else:
             if contain_business_license:
-                file_content.append('BusinessLicense')
+                file_content.append('BUSINESS_LICENSE_IMAGE')
             elif contain_car_license:
-                file_content.append('CarLicense')
+                file_content.append('CAR_LICENSE_IMAGE')
             else:
                 pass