Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -132,7 +132,7 @@ class L2NormLayer(layers.Layer):
|
|
| 132 |
def call(self, inputs):
|
| 133 |
return tf.math.l2_normalize(inputs, axis=self.axis, epsilon=self.epsilon)
|
| 134 |
|
| 135 |
-
class SentenceEncoder(Model):
|
| 136 |
def __init__(self, vocab_size, embed_dim=EMBED_DIM, latent_dim=LATENT_DIM, max_len=MAX_LEN, pad_id=pad_id, dropout_rate=EMBED_DROPOUT):
|
| 137 |
super().__init__()
|
| 138 |
self.pad_id = pad_id
|
|
|
|
| 132 |
def call(self, inputs):
|
| 133 |
return tf.math.l2_normalize(inputs, axis=self.axis, epsilon=self.epsilon)
|
| 134 |
|
| 135 |
+
class SentenceEncoder(tf.keras.Model):
|
| 136 |
def __init__(self, vocab_size, embed_dim=EMBED_DIM, latent_dim=LATENT_DIM, max_len=MAX_LEN, pad_id=pad_id, dropout_rate=EMBED_DROPOUT):
|
| 137 |
super().__init__()
|
| 138 |
self.pad_id = pad_id
|