Spaces:
Runtime error
Runtime error
Мясников Филипп Сергеевич
commited on
Commit
•
5e09808
1
Parent(s):
52e0c74
Fix
Browse files
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 = 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 = '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
|