Мясников Филипп Сергеевич commited on
Commit
90c67b0
1 Parent(s): 5e09808
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ from util import *
34
  from huggingface_hub import hf_hub_download
35
 
36
  device= 'cpu'
37
- model_path_e = 'https://huggingface.co/bankholdup/stylegan_petbreeder/blob/main/e4e_ffhq512_cat.pt' #hf_hub_download(repo_id="bankholdup/stylegan_petbreeder", filename="e4e_ffhq5_cat.pt")
38
  ckpt = torch.load(model_path_e, map_location='cpu')
39
  opts = ckpt['opts']
40
  opts['checkpoint_path'] = model_path_e
 
34
  from huggingface_hub import hf_hub_download
35
 
36
  device= 'cpu'
37
+ model_path_e = hf_hub_download(repo_id="bankholdup/stylegan_petbreeder", filename="e4e_ffhq512_cat.pt")
38
  ckpt = torch.load(model_path_e, map_location='cpu')
39
  opts = ckpt['opts']
40
  opts['checkpoint_path'] = model_path_e