Update app.py

app.py

@@ -6,7 +6,7 @@ def blur_image(image_path, detections, parts_to_blur):
     image = cv2.imread(image_path)
     
     for detection in detections:
-        label = detection['label']
+        label = detection['class']
         if label in parts_to_blur:
             x1, y1, x2, y2 = map(int, detection['box'])
             roi = image[y1:y2, x1:x2]
@@ -22,8 +22,9 @@ def process(input_img):
     detections = detector.detect(input_img)
     print(detections)
     parts_to_blur = [
-        'EXPOSED_GENITALIA_F', 'EXPOSED_GENITALIA_M', 
-        'EXPOSED_BREAST_F', 'EXPOSED_BUTTOCKS_F', 'EXPOSED_BUTTOCKS_M'
+        'FEMALE_GENITALIA_EXPOSED', 'MALE_GENITALIA_EXPOSED', 
+        'FEMALE_BREAST_EXPOSED', 'BUTTOCKS_EXPOSED',
+        'MALE_BREAST_EXPOSED', 'ANUS_EXPOSED'
     ]
     
     blurred_image_path = blur_image(input_img, detections, parts_to_blur)