Update app.py

app.py

@@ -13,6 +13,9 @@ encoded_images = np.load("X_encoded_compressed.npy")
 print("Shape of encoded_images:", encoded_images.shape)
 print("Sample encoded image:", encoded_images[0])
 dataset = load_dataset('eybro/images')
+split_dataset = dataset['train'].train_test_split(test_size=0.2)
+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)