moflo commited on
Commit
d6429ec
1 Parent(s): 445c03a
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -591,7 +591,7 @@ class InferenceWrapper:
591
  tf.random.set_seed(196)
592
  batch_size = 1
593
  self.z = tf.random.normal((batch_size, self.style_gan.z_dim))
594
- self.w = self.style_gan.mapping(z)
595
  self.noise = self.style_gan.generate_noise(batch_size=batch_size)
596
  else:
597
  print(f"Model '{self.model}' already loaded, reusing it.")
 
591
  tf.random.set_seed(196)
592
  batch_size = 1
593
  self.z = tf.random.normal((batch_size, self.style_gan.z_dim))
594
+ self.w = self.style_gan.mapping(self.z)
595
  self.noise = self.style_gan.generate_noise(batch_size=batch_size)
596
  else:
597
  print(f"Model '{self.model}' already loaded, reusing it.")