Update app.py

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import tensorflow as tf
 from tensorflow import keras
 from model import build_siamese
+import cv2
 import os
 
 # Suppress oneDNN warnings (optional)
@@ -59,7 +60,8 @@ stored_imgs = load_stored_images()
 # Inference function
 def check_membership(uploaded_image):
     try:
-        image = uploaded_image.reshape((256,256,3))
+        image = cv2.resize(uploaded_image, (256, 256))
+
         uploaded_image = np.expand_dims(image, axis=0).astype("float32")
 
         predictions = [siamese.predict([uploaded_image, img])[0][0] for img in stored_imgs]