Spaces:

moflo
/

nftGAN

Runtime error

moflo commited on Dec 20, 2021

Commit

fcd4310

1 Parent(s): 7d2e4a6

Typo on pretrained dataset

Files changed (1) hide show

app.py CHANGED Viewed

@@ -587,7 +587,7 @@ class InferenceWrapper:
         self.model = model
         self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
         self.style_gan.grow_model(128)
-        self.style_gan.load_weights(os.path.join("pretrained/stylegan_128x128.ckpt"))
         self.seed = -1
     def __call__(self, seed, feature):

         self.model = model
         self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
         self.style_gan.grow_model(128)
+        self.style_gan.load_weights(os.path.join("pretrained/stylegan_64x64.ckpt"))
         self.seed = -1
     def __call__(self, seed, feature):