Update app.py

app.py

@@ -7,13 +7,38 @@ import os
 # Suppress oneDNN warnings (optional)
 os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
 
-# Load the model
+# Define custom functions
+def euclidean_distance(vectors):
+    x, y = vectors
+    sum_square = tf.reduce_sum(tf.square(x - y), axis=1, keepdims=True)
+    return tf.sqrt(tf.maximum(sum_square, tf.keras.backend.epsilon()))
+
+def contrastive_loss(y_true, y_pred, margin=1.0):
+    square_pred = tf.square(y_pred)
+    margin_square = tf.square(tf.maximum(margin - y_pred, 0))
+    return tf.reduce_mean(y_true * square_pred + (1 - y_true) * margin_square)
+
+# Load model with custom objects
 try:
-    siamese = keras.models.load_model("my_siamese.keras", compile=False)
+    siamese = keras.models.load_model(
+        "my_siamese.keras",
+        custom_objects={
+            "euclidean_distance": euclidean_distance,
+            "contrastive_loss": contrastive_loss
+        },
+        compile=False
+    )
 except Exception as e:
     print(f"Error loading model: {e}")
     raise
 
+# Load the model
+# try:
+#     siamese = keras.models.load_model("my_siamese.keras", compile=False)
+# except Exception as e:
+#     print(f"Error loading model: {e}")
+#     raise
+
 # Load stored images
 def load_stored_images():
     try: