Update app.py

app.py

@@ -25,13 +25,13 @@ from cloob_training import model_pt, pretrained
 import ldm.models.autoencoder
 from diffusion import sampling, utils
 import train_latent_diffusion as train
-from huggingface_hub import hf_hub_url, cached_download
+from huggingface_hub import hf_hub_url, hf_hub_download
 import random
 
 # Download the model files
-checkpoint = cached_download(hf_hub_url("huggan/distill-ccld-wa", filename="model_student.ckpt"))
-ae_model_path = cached_download(hf_hub_url("huggan/ccld_wa", filename="ae_model.ckpt"))
-ae_config_path = cached_download(hf_hub_url("huggan/ccld_wa", filename="ae_model.yaml"))
+checkpoint = hf_hub_download(hf_hub_url("huggan/distill-ccld-wa", filename="model_student.ckpt"))
+ae_model_path = hf_hub_download(hf_hub_url("huggan/ccld_wa", filename="ae_model.ckpt"))
+ae_config_path = hf_hub_download(hf_hub_url("huggan/ccld_wa", filename="ae_model.yaml"))
 
 # Define a few utility functions