Update app.py

app.py

@@ -3,6 +3,7 @@ import numpy as np
 from sklearn.metrics.pairwise import euclidean_distances
 import cv2
 from keras.models import load_model
+from keras.models import Model
 
 autoencoder = load_model("autoencoder_model.keras")
 encoded_images = np.load("X_encoded_compressed.npy")
@@ -41,7 +42,7 @@ def process_image(image):
     img /= 255.0
     img = np.expand_dims(img, axis=0)
 
-    layer_model = Model(inputs=model.input, outputs=model.layers[layer].output)
+    layer_model = Model(inputs=autoencoder.input, outputs=autoencoder.layers[layer].output)
 
     encoded_array = layer_model.predict(img)