Update app.py

app.py

@@ -17,7 +17,6 @@ START = tf.argmax(tf.constant(reserved_tokens) == "[START]")
 END = tf.argmax(tf.constant(reserved_tokens) == "[END]")
 PAD = tf.argmax(tf.constant(reserved_tokens) == "[PAD]")        
 VOCAB_SIZE = len(vocab)
-print(VOCAB_SIZE)
 D_MODEL = 256
 NB_LAYERS = 6
 FFN_UNITS = 2048
@@ -50,6 +49,7 @@ def evaluate(sentence):
     inputs, _ = text.pad_model_inputs(inputs, max_seq_length = MAX_LENGTH + 2, pad_value = PAD)
     
     prediction = transformer(inputs, False)
+    print(prediction)
     
     prediction = tf.round(prediction)
     print(prediction)