fix: correct keys on segmentation

app.py

@@ -5,6 +5,7 @@ import numpy as np
 from PIL import Image
 import random
 from skimage.metrics import structural_similarity as ssim
+import tensorflow as tf
 import requests
 import os
 import transformers
@@ -15,6 +16,7 @@ def set_seed(seed):
     torch.manual_seed(seed)
     torch.cuda.manual_seed_all(seed)
     transformers.set_seed(seed)
+    tf.random.set_seed(seed)
     torch.backends.cudnn.deterministic = True
     torch.backends.cudnn.benchmark = False
 
@@ -81,7 +83,7 @@ def predict_and_compare(image):
         distance = result["distance"]
         
         # Get face area for SSIM
-        face_area = DeepFace.extract_faces(temp_path, detector_backend='retinaface')[0]
+        face_area = DeepFace.extract_faces(temp_path, detector_backend='retinaface')[0]["facial_area"]
         
         # Calculate SSIM against SSIM reference
         ssim_value = compare_face_ssim(image, ssim_reference, face_area)