Update app.py

app.py

@@ -4,9 +4,11 @@ from sklearn.metrics.pairwise import euclidean_distances
 import cv2
 from keras.models import load_model
 from keras.models import Model
+from datasets import load_dataset
 
 autoencoder = load_model("autoencoder_model.keras")
 encoded_images = np.load("X_encoded_compressed.npy")
+dataset = load_dataset('eybro/images')
 
 def find_nearest_neighbors(encoded_images, input_image, top_n=5):
     """