Update app.py

app.py

@@ -10,7 +10,7 @@ from generator import Generator, LATENT_DIM
 import math
 
 generator = Generator(7)
-variables = generator.init(jax.random.PRNGKey(0), jnp.zeros([1, LATENT_DIM]), training=False)
+variables = generator.init(jax.random.PRNGKey(0), jnp.zeros([1, LATENT_DIM]))
 
 fs = HfFileSystem()
 with fs.open("PrakhAI/AIPlane3/g_checkpoint_400000.msgpack", "rb") as f: