Update app.py

app.py

@@ -8,9 +8,7 @@ from huggingface_hub import HfFileSystem
 from flax.serialization import msgpack_restore, from_state_dict
 import time
 from local_response_norm import LocalResponseNorm
-from generator import Generator
-
-LATENT_DIM = 100
+from generator import Generator, LATENT_DIM
 
 generator = Generator()
 variables = generator.init(jax.random.PRNGKey(0), jnp.zeros([1, LATENT_DIM]), training=False)