Мясников Филипп Сергеевич commited on
Commit
d0396da
1 Parent(s): 9dc5640
Files changed (1) hide show
  1. app.py +0 -3
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'):