TomRB22 commited on
Commit
de6190c
1 Parent(s): 133505d

`generate` is now named `decode`

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -184,7 +184,7 @@ class VAE(tf.keras.Model):
184
  z_sample = mu + sd * tf.random.normal(shape=(120,))
185
  return z_sample, mu, sd
186
 
187
- def generate(self, z_sample: tf.Tensor=None) -> tf.Tensor:
188
  """
189
  Decode a latent representation of a song.
190
 
 
184
  z_sample = mu + sd * tf.random.normal(shape=(120,))
185
  return z_sample, mu, sd
186
 
187
+ def decode(self, z_sample: tf.Tensor=None) -> tf.Tensor:
188
  """
189
  Decode a latent representation of a song.
190