OpenLab-NLP commited on
Commit
590e996
·
verified ·
1 Parent(s): c34f761

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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