Corrected error in documentation

model.py

@@ -157,7 +157,7 @@ class VAE(tf.keras.Model):
 
   def encode(self, x_input: tf.Tensor) -> tuple[tf.Tensor]:
     """
-    Get a "song map" and make a forward pass through the encoder, in order
+    Make a forward pass through the encoder for a given song map, in order
     to return the latent representation and the distribution's parameters.
 
     Parameters