Spaces:

keras-io
/

bert-semantic-similarity

Runtime error

vumichien commited on Jul 7, 2022

Commit

bdb4741

1 Parent(s): f11abfb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,7 +41,7 @@ class BertSemanticDataGenerator(tf.keras.utils.Sequence):
         encoded = self.tokenizer.batch_encode_plus(
             sentence_pairs.tolist(),
             add_special_tokens=True,
-            max_length=max_length,
             return_attention_mask=True,
             return_token_type_ids=True,
             pad_to_max_length=True,

         encoded = self.tokenizer.batch_encode_plus(
             sentence_pairs.tolist(),
             add_special_tokens=True,
+            max_length=128,
             return_attention_mask=True,
             return_token_type_ids=True,
             pad_to_max_length=True,