fix: update paths

app.py

@@ -22,11 +22,10 @@ set_seed(42)
 
 flag = os.environ["FLAG"] if "FLAG" in os.environ else "fakeflag{placeholder}"
 
-# Load both reference images
 ssim_reference_url = "https://files.catbox.moe/b6p5fj.png"
 face_reference_url = "https://files.catbox.moe/m0g0ek.png"
-ssim_reference_path = "ssim_reference.jpeg"
-face_reference_path = "face_reference.jpg"
+ssim_reference_path = "ssim_reference.png"
+face_reference_path = "face_reference.png"
 
 # Download SSIM reference
 if not os.path.exists(ssim_reference_path):
@@ -82,8 +81,7 @@ def predict_and_compare(image):
         distance = result["distance"]
         
         # Get face area for SSIM
-        face_detector = DeepFace.build_model('retinaface')
-        face_area = DeepFace.extract_faces(np.array(image), detector_backend='retinaface')[0]
+        face_area = DeepFace.extract_faces(temp_path, detector_backend='retinaface')[0]
         
         # Calculate SSIM against SSIM reference
         ssim_value = compare_face_ssim(image, ssim_reference, face_area)