Testing 64x64

app.py

@@ -560,11 +560,10 @@ TARGET_RES = 128
 
 print("Loading...")
 
-#url = "https://github.com/yarri-oss/stylegan_performance/releases/download/v0.1/stylegan_32x32.ckpt.zip"
-url = "https://github.com/moflo/nftGAN/releases/download/v0.1/stylegan_128x128.ckpt.zip"
+url = "https://github.com/yarri-oss/stylegan_performance/releases/download/v0.1/stylegan_64x64.ckpt.zip"
 
 weights_path = keras.utils.get_file(
-    "stylegan_128x128.ckpt.zip",
+    "stylegan_64x64.ckpt.zip",
     url,
     extract=True,
     cache_dir=os.path.abspath("."),
@@ -590,8 +589,8 @@ class InferenceWrapper:
     def __init__(self, model):
         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.style_gan.grow_model(64)
+        self.style_gan.load_weights(os.path.join("pretrained/stylegan_64x64.ckpt"))
         self.seed = -1
 
     def __call__(self, seed, feature):