eybro commited on
Commit
f7c2be1
·
verified ·
1 Parent(s): 4618782

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import matplotlib.pyplot as plt
11
  autoencoder = load_model("autoencoder_model.keras")
12
  encoded_images = np.load("X_encoded_compressed.npy")
13
 
14
- dataset = load_dataset("eybro/images-split")
15
  split_dataset = dataset['train'].train_test_split(test_size=0.2, seed=42) # 80% train, 20% test
16
  dataset['train'] = split_dataset['train']
17
  dataset['test'] = split_dataset['test']
 
11
  autoencoder = load_model("autoencoder_model.keras")
12
  encoded_images = np.load("X_encoded_compressed.npy")
13
 
14
+ dataset = load_dataset("eybro/images")
15
  split_dataset = dataset['train'].train_test_split(test_size=0.2, seed=42) # 80% train, 20% test
16
  dataset['train'] = split_dataset['train']
17
  dataset['test'] = split_dataset['test']