Spaces:
Runtime error
Runtime error
Мясников Филипп Сергеевич
commited on
Commit
•
d0396da
1
Parent(s):
9dc5640
fix
Browse files
app.py
CHANGED
@@ -53,7 +53,6 @@ ffhq_d_filt = {k[len('decoder') + 1:]: v for k, v in ffhq_ckpt['state_dict'].ite
|
|
53 |
ffhq_decoder.load_state_dict(ffhq_d_filt, strict=True)
|
54 |
ffhq_decoder.eval()
|
55 |
ffhq_decoder.to(device)
|
56 |
-
clear_output()
|
57 |
|
58 |
dog_model_path = hf_hub_download(repo_id="bankholdup/stylegan_petbreeder", filename="e4e_ffhq512_dog.pt")
|
59 |
|
@@ -74,7 +73,6 @@ dog_d_filt = {k[len('decoder') + 1:]: v for k, v in dog_ckpt['state_dict'].items
|
|
74 |
dog_decoder.load_state_dict(dog_d_filt, strict=True)
|
75 |
dog_decoder.eval()
|
76 |
dog_decoder.to(device)
|
77 |
-
clear_output()
|
78 |
|
79 |
cat_model_path = hf_hub_download(repo_id="bankholdup/stylegan_petbreeder", filename="e4e_ffhq512_cat.pt")
|
80 |
|
@@ -95,7 +93,6 @@ cat_d_filt = {k[len('decoder') + 1:]: v for k, v in cat_ckpt['state_dict'].items
|
|
95 |
cat_decoder.load_state_dict(cat_d_filt, strict=True)
|
96 |
cat_decoder.eval()
|
97 |
cat_decoder.to(device)
|
98 |
-
clear_output()
|
99 |
|
100 |
|
101 |
def gen_im(model_type='ffhq'):
|
|
|
53 |
ffhq_decoder.load_state_dict(ffhq_d_filt, strict=True)
|
54 |
ffhq_decoder.eval()
|
55 |
ffhq_decoder.to(device)
|
|
|
56 |
|
57 |
dog_model_path = hf_hub_download(repo_id="bankholdup/stylegan_petbreeder", filename="e4e_ffhq512_dog.pt")
|
58 |
|
|
|
73 |
dog_decoder.load_state_dict(dog_d_filt, strict=True)
|
74 |
dog_decoder.eval()
|
75 |
dog_decoder.to(device)
|
|
|
76 |
|
77 |
cat_model_path = hf_hub_download(repo_id="bankholdup/stylegan_petbreeder", filename="e4e_ffhq512_cat.pt")
|
78 |
|
|
|
93 |
cat_decoder.load_state_dict(cat_d_filt, strict=True)
|
94 |
cat_decoder.eval()
|
95 |
cat_decoder.to(device)
|
|
|
96 |
|
97 |
|
98 |
def gen_im(model_type='ffhq'):
|