Hugging Face's logo

Spaces:
PrakhAI
/
AIPlane3
Sleeping

App Files Community
PrakhAI commited on
Commit
41451f5
·
1 Parent(s): b0990de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import time
9
  from generator import Generator, LATENT_DIM
10
  import math
11
 
12
- generator = Generator()
13
  variables = generator.init(jax.random.PRNGKey(0), jnp.zeros([1, LATENT_DIM]), training=False)
14
 
15
  fs = HfFileSystem()
 
9
  from generator import Generator, LATENT_DIM
10
  import math
11
 
12
+ generator = Generator(7)
13
  variables = generator.init(jax.random.PRNGKey(0), jnp.zeros([1, LATENT_DIM]), training=False)
14
 
15
  fs = HfFileSystem()