Update app.py

app.py

@@ -11,6 +11,11 @@ model.eval()
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 
+PAD_TOKEN = "[PAD]"
+MASK_TOKEN = "[MASK]"
+
+vocab = {PAD_TOKEN: 0, MASK_TOKEN: 1}
+
 # Assume vocab, id_to_word, pad_id, mask_id already defined
 
 # =====================