tombio commited on
Commit
cd308ea
1 Parent(s): 42b1267

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,8 +28,8 @@ import numpy as np
28
  #pathlib.PosixPath = pathlib.WindowsPath
29
 
30
  from huggingface_hub import hf_hub_download
31
- ckpt = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-pruned.ckpt", cache_dir="/data/.cache")
32
- config = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-config.yaml", cache_dir="/data/.cache")
33
 
34
  device = "cuda:0"
35
  model = load_model_from_config(config, ckpt, device=device, verbose=False)
 
28
  #pathlib.PosixPath = pathlib.WindowsPath
29
 
30
  from huggingface_hub import hf_hub_download
31
+ ckpt = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-pruned.ckpt")
32
+ config = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-config.yaml")
33
 
34
  device = "cuda:0"
35
  model = load_model_from_config(config, ckpt, device=device, verbose=False)