Update app.py

app.py

@@ -61,6 +61,7 @@ class BertSemanticDataGenerator(tf.keras.utils.Sequence):
             return [input_ids, attention_masks, token_type_ids]
 
 model = from_pretrained_keras("keras-io/bert-semantic-similarity")
+labels = ["contradiction", "entailment", "neutral"]
 
 def predict(sentence1, sentence2):
     sentence_pairs = np.array([[str(sentence1), str(sentence2)]])