Update app.py

app.py

@@ -1,9 +1,19 @@
+import tensorflow as tf
+from tensorflow.keras.layers import Layer
 from tensorflow.keras.models import load_model
 
-# Replace 'final_siamese_network.h5' with the path to your saved model
-loaded_siamese_network = load_model('siamese_network.h5')
+# Define the custom layer
+class DistanceLayer(Layer):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
 
-loaded_siamese_network.summary()
+    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)
 
-print("Input shape:", loaded_siamese_network.input_shape)
-print("Output shape:", loaded_siamese_network.output_shape)
+# 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()