moflo commited on
Commit
e6dd44f
·
1 Parent(s): fcd4310

Typo on pretrained download

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -563,7 +563,7 @@ print("Loading...")
563
  url = "https://github.com/yarri-oss/stylegan_performance/releases/download/v0.1/stylegan_64x64.ckpt.zip"
564
 
565
  weights_path = keras.utils.get_file(
566
- "stylegan_64x64.ckpt.zip",
567
  url,
568
  extract=True,
569
  cache_dir=os.path.abspath("."),
@@ -586,7 +586,7 @@ class InferenceWrapper:
586
  def __init__(self, model):
587
  self.model = model
588
  self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
589
- self.style_gan.grow_model(128)
590
  self.style_gan.load_weights(os.path.join("pretrained/stylegan_64x64.ckpt"))
591
  self.seed = -1
592
 
 
563
  url = "https://github.com/yarri-oss/stylegan_performance/releases/download/v0.1/stylegan_64x64.ckpt.zip"
564
 
565
  weights_path = keras.utils.get_file(
566
+ "stylegan_64x64.ckpt",
567
  url,
568
  extract=True,
569
  cache_dir=os.path.abspath("."),
 
586
  def __init__(self, model):
587
  self.model = model
588
  self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
589
+ self.style_gan.grow_model(64)
590
  self.style_gan.load_weights(os.path.join("pretrained/stylegan_64x64.ckpt"))
591
  self.seed = -1
592