Update app.py

app.py

@@ -1,19 +1,10 @@
 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 custom layer
-class DistanceLayer(Layer):
-    def __init__(self, **kwargs):
-        super().__init__(**kwargs)
+# Define the target shape as previously used
+target_shape = (160, 160)
 
-    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()
+# Load the FaceNet model with custom weights
+facenet_model = load_model('facenet_keras.h5', compile=False)