New checkpoint and pixelation

app.py

@@ -560,7 +560,7 @@ TARGET_RES = 128
 
 print("Loading...")
 
-url = "https://github.com/yarri-oss/stylegan_performance/releases/download/v0.1/stylegan_128x128.ckpt.zip"
+url = "https://github.com/moflo/nftGAN/releases/download/v0.1/stylegan_128x128.ckpt.zip"
 
 weights_path = keras.utils.get_file(
     "stylegan_64x64.ckpt.zip",
@@ -609,6 +609,7 @@ class InferenceWrapper:
         res = 64
         image = tf.image.resize(image, (res, res), method=tf.image.ResizeMethod.NEAREST_NEIGHBOR)
         image = tf.cast(image, tf.float32) / 127.5 - 1.0
+        image = np.clip((image * 0.5 + 0.5) * 255, 0, 255).astype(np.uint8)
 
         return image