eybro commited on
Commit
ffbab7b
·
verified ·
1 Parent(s): bd24987

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ import matplotlib.pyplot as pl
10
  from huggingface_hub import hf_hub_download
11
 
12
  # Download and load model and encoded images
13
- model_path = hf_hub_download(repo_id="eybro/autoencoder", filename="autoencoder_model.keras")
14
- data_path = hf_hub_download(repo_id="eybro/encoded_images", filename="X_encoded_compressed.npy")
15
 
16
  autoencoder = load_model(model_path)
17
  encoded_images = np.load(data_path)
 
10
  from huggingface_hub import hf_hub_download
11
 
12
  # Download and load model and encoded images
13
+ model_path = hf_hub_download(repo_id="eybro/autoencoder", filename="autoencoder_model.keras", repo_type='model')
14
+ data_path = hf_hub_download(repo_id="eybro/encoded_images", filename="X_encoded_compressed.npy", repo_type='dataset')
15
 
16
  autoencoder = load_model(model_path)
17
  encoded_images = np.load(data_path)