Update app.py

app.py

@@ -11,6 +11,11 @@ with open("vocab.txt", mode = "r", encoding = "utf-8") as file:
     for token in file:
         vocab.append(token.replace("\n", ""))
         
+reserved_tokens=["[START]", "[END]", "[PAD]", "[UNK]"]
+        
+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)
 D_MODEL = 256
 NB_LAYERS = 6