Update app.py

app.py

@@ -59,7 +59,8 @@ stored_imgs = load_stored_images()
 # Inference function
 def check_membership(uploaded_image):
     try:
-        uploaded_image = np.expand_dims(uploaded_image, axis=0).astype("float32")
+        image = uploaded_image.reshape((256,256,3))
+        uploaded_image = np.expand_dims(image, axis=0).astype("float32")
 
         predictions = [siamese.predict([uploaded_image, img])[0][0] for img in stored_imgs]