Update app.py

app.py

@@ -16,8 +16,8 @@ import clip
 from PIL import Image
 
 from huggingface_hub import hf_hub_download
-ckpt = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-pruned.ckpt")
-config = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-config.yaml")
+ckpt = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-pruned.ckpt", cache_dir="/data/.cache")
+config = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-config.yaml", cache_dir="/data/.cache")
 
 device = "cuda:0"
 model = load_model_from_config(config, ckpt, device=device, verbose=False)