Update app.py

app.py

@@ -1,10 +1,19 @@
 import tensorflow as tf
+from tensorflow.keras.layers import Layer
 from tensorflow.keras.models import load_model
-import numpy as np
-from PIL import Image
 
-# Define the target shape as previously used
-target_shape = (160, 160)
+# Define the custom layer
+class DistanceLayer(Layer):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
 
-# Load the FaceNet model with custom weights
-facenet_model = load_model('facenet_keras.h5', compile=False)
+    def call(self, anchor, positive, negative):
+        ap_distance = tf.reduce_sum(tf.square(anchor - positive), -1)
+        an_distance = tf.reduce_sum(tf.square(anchor - negative), -1)
+        return (ap_distance, an_distance)
+
+# Load the model with the custom layer
+siamese_network = load_model('siamese_network.h5', custom_objects={'DistanceLayer': DistanceLayer})
+
+# Now, you can use siamese_network for predictions, evaluation, etc.
+siamese_network.summary()