Spaces:

bankholdup
/

stylegan_petbreeder

Runtime error

Мясников Филипп Сергеевич commited on May 18, 2022

Commit

90c67b0

•

1 Parent(s): 5e09808

Fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,7 +34,7 @@ from util import *
 from huggingface_hub import hf_hub_download
 device= 'cpu'
-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")
 ckpt = torch.load(model_path_e, map_location='cpu')
 opts = ckpt['opts']
 opts['checkpoint_path'] = model_path_e

 from huggingface_hub import hf_hub_download
 device= 'cpu'
+model_path_e = hf_hub_download(repo_id="bankholdup/stylegan_petbreeder", filename="e4e_ffhq512_cat.pt")
 ckpt = torch.load(model_path_e, map_location='cpu')
 opts = ckpt['opts']
 opts['checkpoint_path'] = model_path_e