Update app.py

app.py

@@ -12,6 +12,9 @@ autoencoder = load_model("autoencoder_model.keras")
 encoded_images = np.load("X_encoded_compressed.npy")
 
 dataset = load_dataset("eybro/images-split")
+split_dataset = dataset['train'].train_test_split(test_size=0.2, seed=42)  # 80% train, 20% test
+dataset['train'] = split_dataset['train']
+dataset['test'] = split_dataset['test']
 
 num_clusters = 10  # Choose the number of clusters
 kmeans = KMeans(n_clusters=num_clusters, random_state=42)