update path info

app.py

@@ -571,7 +571,7 @@ weights_path = keras.utils.get_file(
 )
 
 print("Pretrained:")
-print(os.listdir(os.path.join("pretrained")))
+print(os.listdir(os.path.join("pretrained/checkpoints")))
 
 # style_gan.grow_model(128)
 # style_gan.load_weights(os.path.join("pretrained/stylegan_128x128.ckpt"))
@@ -590,7 +590,7 @@ class InferenceWrapper:
         self.model = model
         self.style_gan = StyleGAN(start_res=START_RES, target_res=TARGET_RES)
         self.style_gan.grow_model(64)
-        self.style_gan.load_weights(os.path.join("pretrained/stylegan_64x64.ckpt"))
+        self.style_gan.load_weights(os.path.join("pretrained/checkpoints/stylegan_64x64.ckpt"))
         self.seed = -1
 
     def __call__(self, seed, feature):